fix bug for dataset dir. (issue #3)

ZhangGongjie · Apr 15, 2022 · 16d79d3 · 16d79d3
1 parent c2e6a8f
commit 16d79d3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/datasets/coco.py b/datasets/coco.py
@@ -151,9 +151,9 @@ def build(image_set, args):
     assert root.exists(), f'provided COCO path {root} does not exist'
     mode = 'instances'
     PATHS = {
-        "train": (root / "images" / "train2017", root / "annotations" / f'{mode}_train2017.json'),
-        "val": (root / "images" / "val2017", root / "annotations" / f'{mode}_val2017.json'),
-        "test": (root / "images" / "test2017", root / "annotations" / f'image_info_test-dev2017.json'),
+        "train": (root / "train2017", root / "annotations" / f'{mode}_train2017.json'),
+        "val": (root / "val2017", root / "annotations" / f'{mode}_val2017.json'),
+        "test": (root / "test2017", root / "annotations" / f'image_info_test-dev2017.json'),
     }
 
     img_folder, ann_file = PATHS[image_set]