diff --git a/valcoco.py b/valcoco.py new file mode 100644 index 0000000..d370f32 --- /dev/null +++ b/valcoco.py @@ -0,0 +1,16 @@ +import sys +from pycocotools.coco import COCO +from pycocotools.cocoeval import COCOeval + + +if __name__ == '__main__': + pred = sys.argv[1] + gt = 'instances_val2017.json' + + cocoGt = COCO(gt) + cocoDt = cocoGt.loadRes(pred) + + cocoEval = COCOeval(cocoGt, cocoDt, 'bbox') + cocoEval.evaluate() + cocoEval.accumulate() + cocoEval.summarize()