From 2b33e56d4e73d548be352b5582b3da411e975d07 Mon Sep 17 00:00:00 2001 From: wat3rBro Date: Fri, 14 Dec 2018 02:40:01 -0800 Subject: [PATCH] use empty COCO to avoid loading empty result via coco api (#270) --- maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py b/maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py index 9690c0d..2fe503d 100644 --- a/maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py +++ b/maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py @@ -277,9 +277,11 @@ def evaluate_predictions_on_coco( 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)) + 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.evaluate() -- GitLab