
    |-j                     F    d dl mZ ddlmZ ddlmZ  G d de          ZdS )	    )Path   )BaseEvaluator   )MODELSc                   (    e Zd ZdZeZd ZdefdZdS )TextRecEvaluatorz Text Recognition Model Evaluatorc                    | j         j        r$| j                            | j         j                   | j        d         dk    r&| j                            | j        j        d           nZd| j        d         v r&| j                            | j        j        d           n%| j                            | j        j        d           d}| j                             d          r| j         j        }n7t          | j         j
                  j        d	z  }|                                sd}|| j                            |           dS dS )
zupdate evaluation configmodelLaTeX_OCR_recLaTeXOCRDataSetzPP-OCRv3SimpleDataSetMSTextRecDatasetNlabel_dict_pathzlabel_dict.txt)eval_configlog_interval
pdx_configupdate_log_intervalglobal_configupdate_datasetdataset_dirgetr   r   weight_pathparentexistsupdate_label_dict_path)selfr   s     j/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/modules/text_recognition/evaluator.pyupdate_configzTextRecEvaluator.update_config   sf   ( 	OO//0@0MNNNg&/99O**".0A    4-g666O**".    O**".0B    122 	'".>OO T%1229<LL  #))++ '"&&O22?CCCCC '&    returnc                 D    | j         j        |                                 dS )zget key-value arguments of model evaluation function

        Returns:
            dict: the arguments of evaluation function.
        )r   device)r   r   
get_device)r   s    r   get_eval_kwargsz TextRecEvaluator.get_eval_kwargs7   s)      +7oo''
 
 	
r    N)	__name__
__module____qualname____doc__r   entitiesr   dictr%    r    r   r	   r	      sL        **HD D D8	
 	
 	
 	
 	
 	
 	
r    r	   N)pathlibr   baser   
model_listr   r	   r,   r    r   <module>r0      sq                            *
 *
 *
 *
 *
} *
 *
 *
 *
 *
r    