o
    yqi0                     @   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 )InstanceSegTrainerz#Instance Segmentation Model Trainerc                 C   s   | j j| jjdg dd dS )zupdate dataset settingsZCOCOInstSegDataset)imageZgt_bboxZgt_classZgt_polyZis_crowd)Zdata_fieldsN)Z
pdx_configZupdate_datasetZglobal_configZdataset_dir)self r   t/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddlex/modules/instance_segmentation/trainer.py_update_dataset   s
   
z"InstanceSegTrainer._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   