From 7a01223db47bd9d1b714e8f6a1a94f7db5853bf1 Mon Sep 17 00:00:00 2001 From: jario-jin Date: Fri, 11 Aug 2023 11:27:38 +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 9412ddd..7ff2303 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/gt_coco.json' + path = os.path.abspath(os.path.join(os.getcwd(),"../../..")) + pred_json = 'pd_coco.json' + anno_json = path + '/val2017/gt_coco.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))