o 1jR @s8ddlmZddlmZddlmZGdddeZdS) )Path) BaseEvaluator)MODELSc@s*eZdZdZeZddZdefddZdS)FormulaRecEvaluatorz Text Recognition Model EvaluatorcCs|jjr |j|jj|jddkr|j|jjdn|jddvr-|j|jjdd}|jdr:|jj}nt |jj j d}| sId}|durS|j ||jjduru|jddkrk|jj|jjd d n |jj|jjd d |jd ddur|jj|jjd d dSdS) zupdate evaluation configmodelZ LaTeX_OCR_recZLaTeXOCRDataSet)Z UniMERNetzPP-FormulaNet-LzPP-FormulaNet-SzPP-FormulaNet_plus-LzPP-FormulaNet_plus-MzPP-FormulaNet_plus-SZ SimpleDataSetNlabel_dict_pathzlabel_dict.txteval)mode delimiter) eval_configZ log_intervalZ pdx_configZupdate_log_intervalZ global_configZupdate_datasetZ dataset_dirgetr r weight_pathparentexistsZupdate_label_dict_pathZ batch_sizeZupdate_batch_size_pairZupdate_batch_sizeZupdate_delimiterr )selfr rn/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/formula_recognition/evaluator.py update_configs<    z!FormulaRecEvaluator.update_configreturncCs|jj|dS)zget key-value arguments of model evaluation function Returns: dict: the arguments of evaluation function. )rZdevice)r rZ get_device)rrrrget_eval_kwargsGsz#FormulaRecEvaluator.get_eval_kwargsN) __name__ __module__ __qualname____doc__rentitiesrdictrrrrrrs ,rN)pathlibrbaserZ model_listrrrrrrs