o
    1j                     @   s,   d dl mZ ddlmZ G dd deZdS )   )BaseEvaluator   )MODELSc                   @   s*   e Zd ZdZeZdd ZdefddZdS )VideoDetEvaluatorz$Image Classification Model Evaluatorc                 C   s>   | j jr| j| j j | j| jjd | j| j j dS )zupdate evaluation configZVideoDetDatasetN)	eval_configZlog_intervalZ
pdx_configZupdate_log_intervalZupdate_datasetZglobal_configZdataset_dirZupdate_pretrained_weightsweight_pathself r
   j/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/video_detection/evaluator.pyupdate_config   s   zVideoDetEvaluator.update_configreturnc                 C   s   | j j| jdddS )zget key-value arguments of model evaluation function

        Returns:
            dict: the arguments of evaluation function.
        r   )Zusing_device_number)r   Zdevice)r   r   Z
get_devicer   r
   r
   r   get_eval_kwargs!   s   
z!VideoDetEvaluator.get_eval_kwargsN)	__name__
__module____qualname____doc__r   entitiesr   dictr   r
   r
   r
   r   r      s
    	r   N)baser   Z
model_listr   r   r
   r
   r
   r   <module>   s   