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))