o
    1j                     @   s,   d dl mZ ddlmZ G dd deZdS )   )
DetTrainer   )MODELSc                   @   s   e Zd ZdZeZdd ZdS )LayoutAnalysisTrainerzLayout Analysis Model Trainerc                 C   s6   d| j v r	| j jnd}| j j| jjdg d|d dS )zupdate dataset settingsmetricZCOCOZCOCOInstSegDataset)imageZgt_bboxZgt_classZgt_polyZis_crowdZgt_read_order)Zdata_fieldsr   N)Z
pdx_configr   Zupdate_datasetZglobal_configZdataset_dir)selfr    r	   h/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/layout_analysis/trainer.py_update_dataset   s   
z%LayoutAnalysisTrainer._update_datasetN)__name__
__module____qualname____doc__r   entitiesr   r	   r	   r	   r
   r      s    r   N)Zobject_detectionr   Z
model_listr   r   r	   r	   r	   r
   <module>   s   