diff --git a/datasets/mscoco.py b/datasets/mscoco.py index 43405fe..e79d95c 100644 --- a/datasets/mscoco.py +++ b/datasets/mscoco.py @@ -113,7 +113,7 @@ def _parse_image_path(self, entry): """ dirname, filename = entry['coco_url'].split('/')[-2:] - abs_path = os.path.join(self.root, dirname, filename) + abs_path = os.path.join(self.root, 'images', dirname, filename) return abs_path def __len__(self):