o
    )j                     @   s  d dl mZmZmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZmZmZ edZG dd deZi e	jejge	jejge	jejge	jejge	jejge	jejge	jejge	jejge	jejge	jejge	j ej!ge	j"ej#ge	j$ej%ge	j&ej'ge	j(ej)ge	j*ej)ge	j+ej,ge	j-ej.ge	j/ej0ge	j1ej2ge	j3ej4ge	j5ej6ge	j7ej8ge	j9ej)ge	j:ej;ge	j<ej=ge	j>ej?gi
Z@ed	fd
eeAef deAdeBfddZCd	S )    )DictMappingUnion)Metrics)
ConfigDict)Tasks)Registrybuild_from_cfgdefault_groupZmetricsc                   @   s   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"S )#
MetricKeysaccuracyf1z	binary-f1zmacro-f1zmicro-f1	precisionZrecallZpsnrZssimZlpipsZniqeZavg_lossZfscoreZfidzbleu-1zbleu-4zrouge-1zrouge-lZnedmAPZinbatch_t2i_recall_at_1Zcropping_ratioZdistortion_valueZstability_scoreZpplZplccZsrccZrmseZmrrZndcgARZcolorfulnessZkendall_tau_correlationN)$__name__
__module____qualname__ZACCURACYZF1Z	Binary_F1ZMacro_F1ZMicro_F1Z	PRECISIONZRECALLZPSNRZSSIMZLPIPSZNIQEZAVERAGE_LOSSZFScoreZFIDZBLEU_1ZBLEU_4ZROUGE_1ZROUGE_LZNEDr   ZBatchAccZCROPPING_RATIOZDISTORTION_VALUEZSTABILITY_SCOREZPPLZPLCCZSRCCZRMSEZMRRZNDCGr   ZColorfulnessZKendall_Tau_Correlation r   r   [/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/metrics/builder.pyr      sD    r   N
metric_cfgfielddefault_argsc                 C   s4   t | trd| v sJ ntd| i} t| t||dS )a/   Build metric given metric_name and field.

    Args:
        metric_name (str | dict): The metric name or metric config dict.
        field (str, optional):  The field of this metric, default value: 'default' for all fields.
        default_args (dict, optional): Default initialization arguments.
    type)Z	group_keyr   )
isinstancer   r   r	   METRICS)r   r   r   r   r   r   build_metricT   s   

r   )Dtypingr   r   r   Zmodelscope.metainfor   Zmodelscope.utils.configr   Zmodelscope.utils.constantr   Zmodelscope.utils.registryr   r	   r
   r   objectr   Zimage_segmentationZimage_ins_seg_coco_metricZsentence_similarityZseq_cls_metricZnliZsentiment_classificationZtoken_classificationZtoken_cls_metricZtext_generationZtext_gen_metricZchatZtext_classificationZimage_denoisingZimage_denoise_metricZimage_deblurringZvideo_super_resolutionZvideo_super_resolution_metricZimage_color_enhancementZimage_color_enhance_metricZimage_portrait_enhancementZ!image_portrait_enhancement_metricZvideo_summarizationZvideo_summarization_metricZimage_captioningr   Zvisual_question_answeringZmovie_scene_segmentationZmovie_scene_segmentation_metricZimage_inpaintingZimage_inpainting_metricZ#referring_video_object_segmentationZ*referring_video_object_segmentation_metricZvideo_frame_interpolationZ video_frame_interpolation_metricZvideo_stabilizationZvideo_stabilization_metricZ$image_quality_assessment_degradationZ+image_quality_assessment_degradation_metricZimage_quality_assessment_mosZ#image_quality_assessment_mos_metricZbad_image_detectingZocr_recognitionZocr_recognition_metricZefficient_diffusion_tuningZloss_metricZtranslation_evaluationZtranslation_evaluation_metricZtask_default_metricsstrdictr   r   r   r   r   <module>   s   $








	













%