o 1j @sddlZddlZddlZddlmZmZedr$ddlmZddlm Z ddZ d d Z ed    dd dZ e dkrGe Ze edSdS)N)function_requires_depsis_dep_availableZ pycocotools)COCO)COCOevalcCs4t}|jdtdd|jdtdd|}|S)zParse input argumentsz--prediction_json_pathz ./bbox.json)typedefaultz--gt_json_pathz./instance_val.json)argparseArgumentParser add_argumentstr parse_args)parserargsre/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/base/utils/coco_eval.pyr s r cCs8|j}|j}tj|sJd|t|d|ddS)z4 cocoapi eval with already exists bbox.json z$The json directory:{} does not existbbox) anno_fileN)prediction_json_path gt_json_pathospathexistsformat cocoapi_eval)rrrrrrjson_eval_results%srdi,iTc Cs|dus |dus J|durt|}||}|dkr,t||d}d|j_t||j_n|dkr9t|||||}nt|||}||| t j |j S)a Args: jsonfile (str): Evaluation json file, eg: bbox.json style (str): COCOeval style, can be `bbox` coco_gt (str): Whether to load COCOAPI through anno_file, eg: coco_gt = COCO(anno_file) anno_file (str): COCO annotations file. max_dets (tuple): COCO evaluation maxDets. sigmas (nparray): keypoint labelling sigmas. use_area (bool): If gt annotations (eg. CrowdPose, AIC) do not have 'area', please set use_area=False. NZproposalrrZkeypoints_crowd)rZloadResrparamsZuseCatslistZmaxDetsevaluate accumulateZ summarizesysstdoutflushstats) ZjsonfilestyleZcoco_gtrZmax_detsZsigmasZuse_areaZcoco_dtZ coco_evalrrrr1s     r__main__)NNrNT)r rr"Z utils.depsrrZpycocotools.cocorZpycocotools.cocoevalrr rr__name__rrrrrs(     *