Я использую Python Coco API для запуска оценки для обнаружения объектов. У меня есть два файла, основную правду JSON и результаты JSON. coco notebook demo показывает только запуск eval для всех классов. Как я могу запустить его только для одного определенного класса или подмножества классов? В настоящее время я делаю это:
from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval
annType = 'bbox'
cocoGt=COCO(gt_json)
cocoDt=cocoGt.loadRes(results_json)
imgIds=sorted(cocoGt.getImgIds())
# running evaluation
cocoEval = COCOeval(cocoGt,cocoDt,annType)
cocoEval.params.imgIds = imgIds
cocoEval.evaluate()
cocoEval.accumulate()
cocoEval.summarize()