o 1j@s@ddlZddlmZddlmZddlmZGdddeZdS) N)Path) BaseEvaluator)MODELSc@s2eZdZdZeZddZddZdefddZ d S) SegEvaluatorz%Semantic Segmentation Model EvaluatorcCs&|j|jjd|jjddddS)zupdate evaluation configZ SegDatasetNT)Z is_backbone)Z pdx_configZupdate_datasetZ global_configZ dataset_dirZupdate_pretrained_weights)selfr p/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/semantic_segmentation/evaluator.py update_configszSegEvaluator.update_configcCst|jjd}|S)z get config path Args: weight_path (str): The path to the weight Returns: config_path (str): The path to the config z config.yaml)rparent)r weight_pathZ config_pathr r r get_config_path!s zSegEvaluator.get_config_pathreturncCs$|}tjdd|jj|dS)zget key-value arguments of model evaluation function Returns: dict: the arguments of evaluation function. ZFLAGS_npu_jit_compileN)r device)Z get_deviceosenvironpopZ eval_configr )rrr r r get_eval_kwargs1szSegEvaluator.get_eval_kwargsN) __name__ __module__ __qualname____doc__rentitiesr rdictrr r r r rs r)rpathlibrbaserZ model_listrrr r r r s