From 5c30296b734c1a6fa4ed74b664f411469e17e4f4 Mon Sep 17 00:00:00 2001 From: jario-jin Date: Mon, 14 Aug 2023 01:53:16 +0000 Subject: [PATCH] update samples/test/eval_mAP_on_coco_val/coco_eval.py. Signed-off-by: jario-jin --- .../test/eval_mAP_on_coco_val/coco_eval.py | 30 +++++++++---------- 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/samples/test/eval_mAP_on_coco_val/coco_eval.py b/samples/test/eval_mAP_on_coco_val/coco_eval.py index 81391d5..b534d29 100644 --- a/samples/test/eval_mAP_on_coco_val/coco_eval.py +++ b/samples/test/eval_mAP_on_coco_val/coco_eval.py @@ -3,23 +3,23 @@ from pycocotools.cocoeval import COCOeval import os if __name__ == '__main__': - path = os.path.abspath(os.path.join(os.getcwd(),"../../..")) - pred_json = 'pd_coco.json' - anno_json = path + '/val2017/instances_val2017.json' + path = os.path.abspath(os.path.join(os.getcwd(),"../../..")) + pred_json = 'pd_coco.json' + anno_json = path + '/val2017/instances_val2017.json' - # use COCO API to load forecast results and annotations - cocoGt = COCO(anno_json) - cocoDt = cocoGt.loadRes(pred_json) + # use COCO API to load forecast results and annotations + cocoGt = COCO(anno_json) + cocoDt = cocoGt.loadRes(pred_json) - # create COCO eval object - cocoEval = COCOeval(cocoGt, cocoDt,'bbox') + # create COCO eval object + cocoEval = COCOeval(cocoGt, cocoDt,'bbox') - # assessment - cocoEval.evaluate() - cocoEval.accumulate() - cocoEval.summarize() + # assessment + cocoEval.evaluate() + cocoEval.accumulate() + cocoEval.summarize() - # save results - with open('coco_eval.txt', 'w') as f: - f.write(str(cocoEval.stats)) + # save results + with open('coco_eval.txt', 'w') as f: + f.write(str(cocoEval.stats))