
    9i}                     2    S SK Jr  SSKJr   " S S\5      rg)   )BaseEvaluator   )MODELSc                   6    \ rS rSrSr\rS rS rS\	4S jr
Srg)	DetEvaluator   z Object Detection Model Evaluatorc                    SU R                   ;   a  U R                   R                  OSnSU R                   R                  ;   a  U R                   R                  S   OSnU R                   R                  U R                  R
                  SUUS9  g)zupdate dataset settingsmetricCOCOdata_fieldsNCOCODetDataset)r   r
   )
pdx_configr
   EvalDatasetupdate_datasetglobal_configdataset_dir)selfr
   r   s      j/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/modules/object_detection/evaluator.py_update_datasetDetEvaluator._update_dataset   s    +3t+F''F  ; ;; OO''6 	 	&&**#	 	' 	
    c                    U R                   R                  (       a/  U R                  R                  U R                   R                  5        U R	                  5         U R                  R                  U R                   R                  5        g)zupdate evaluation configN)eval_configlog_intervalr   update_log_intervalr   update_weightsweight_pathr   s    r   update_configDetEvaluator.update_config)   sY    ((OO//0@0@0M0MN&&t'7'7'C'CDr   returnc                 N    U R                   R                  U R                  SS9S.$ )zoget key-value arguments of model evaluation function

Returns:
    dict: the arguments of evaluation function.
r   )using_device_number)r   device)r   r   
get_devicer   s    r   get_eval_kwargsDetEvaluator.get_eval_kwargs0   s+      ++77oo!o<
 	
r    N)__name__
__module____qualname____firstlineno____doc__r   entitiesr   r   dictr&   __static_attributes__r(   r   r   r   r      s!    *H
 E	
 	
r   r   N)baser   
model_listr   r   r(   r   r   <module>r3      s     ! %
= %
r   