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 7ff2303..81391d5 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/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)) diff --git a/samples/test/eval_mAP_on_coco_val/eval_mAP_on_coco_val.cpp b/samples/test/eval_mAP_on_coco_val/eval_mAP_on_coco_val.cpp index dc31f8a..5ec4514 100644 --- a/samples/test/eval_mAP_on_coco_val/eval_mAP_on_coco_val.cpp +++ b/samples/test/eval_mAP_on_coco_val/eval_mAP_on_coco_val.cpp @@ -65,7 +65,7 @@ int main(int argc, char *argv[]) //create pred file std::string val_image_name = GetImageFileName(val_image[i]); - std::string filename = folder+ val_image_name + ".txt"; + std::string filename = folder+"/"+ val_image_name + ".txt"; std::ofstream file(filename); file.is_open(); file<