def evaluate_predictions_on_coco(coco_gt, coco_results, json_result_file, iou_type="bbox", use_cat_label=True): import json with open(json_result_file, "w") as f: json.dump(coco_results, f) from pycocotools.coco import COCO from pycocotools.cocoeval import COCOeval coco_dt = coco_gt.loadRes( str(json_result_file)) if coco_results else COCO() # coco_dt = coco_gt.loadRes(coco_results) coco_eval = COCOeval(coco_gt, coco_dt, iou_type) coco_eval.useCats = use_cat_label coco_eval.evaluate() coco_eval.accumulate() coco_eval.summarize() return coco_eval