COCO annotations JSON fix (#3764)

ultralytics · Jun 24, 2021 · 03281f8 · 03281f8
1 parent f79d747
commit 03281f8
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: