COCO annotations JSON fix (ultralytics#3764)

fhkiel-mlaip · Jun 24, 2021 · 39130f3 · 39130f3
1 parent a5f7e25
commit 39130f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test.py b/test.py
@@ -270,7 +270,7 @@ def run(data,
     # Save JSON
     if save_json and len(jdict):
         w = Path(weights[0] if isinstance(weights, list) else weights).stem if weights is not None else ''  # weights
-        anno_json = '../coco/annotations/instances_val2017.json'  # annotations json
+        anno_json = str(Path(data.get('path', '../coco')) / 'annotations/instances_val2017.json')  # annotations json
         pred_json = str(save_dir / f"{w}_predictions.json")  # predictions json
         print('\nEvaluating pycocotools mAP... saving %s...' % pred_json)
         with open(pred_json, 'w') as f: