o
    1j                     @   s,   d dl mZ ddlmZ G dd deZdS )   )DetEvaluator   )MODELSc                   @   s   e Zd ZdZeZdd ZdS )KeypointEvaluatorz Object Detection Model Evaluatorc                 C   sh   | j jr| j| j j | jj}d| jjv r| jjd nd}| jj| jjd||d | j	| j j
 dS )zupdate evaluation configdata_fieldsNZKeypointTopDownCocoDataset)r   metric)Zeval_configZlog_intervalZ
pdx_configZupdate_log_intervalr   ZTrainDatasetZupdate_datasetZglobal_configZdataset_dirZupdate_weightsZweight_path)selfr   r    r	   m/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/keypoint_detection/evaluator.pyupdate_config   s   zKeypointEvaluator.update_configN)__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   