fix Coco implementation

pytorch · Feb 23, 2021 · e2b6938 · e2b6938
1 parent 315c4f6
commit e2b6938
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/torchvision/datasets/coco.py b/torchvision/datasets/coco.py
@@ -30,7 +30,11 @@ def __init__(
         super().__init__(root, transforms, transform, target_transform)
         from pycocotools.coco import COCO
 
-        self.coco = COCO(annFile)
+        try:
+            self.coco = COCO(annFile)
+        except FileNotFoundError as error:
+            raise RuntimeError(f"The file {annFile} does not exist or is corrupt.") from error
+
         self.ids = list(sorted(self.coco.imgs.keys()))
 
     def _load_image(self, id: int) -> Image.Image: