[fix] Some imports and add check when PIL cannot read images from a f…

…older
facebookresearch · May 9, 2020 · 163daea · 163daea
1 parent 0e619eb
commit 163daea
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/grid_feats/__init__.py b/grid_feats/__init__.py
@@ -3,6 +3,7 @@
 from .build_loader import (
     build_detection_train_loader_with_attributes,
     build_detection_test_loader_with_attributes,
+    build_detection_test_loader_for_images,
 )
 from .roi_heads import AttributeRes5ROIHeads, AttributeStandardROIHeads
 from . import visual_genome
diff --git a/grid_feats/build_loader.py b/grid_feats/build_loader.py
@@ -11,7 +11,7 @@
 from detectron2.data.build import get_detection_dataset_dicts, worker_init_reset_seed, trivial_batch_collator
 from detectron2.data.common import AspectRatioGroupedDataset, DatasetFromList, MapDataset
 
-from .dataset_mapper import AttributeDatasetMapper
+from .dataset_mapper import AttributeDatasetMapper, TestDatasetMapper
 
 
 def build_detection_train_loader_with_attributes(cfg, mapper=None):

diff --git a/grid_feats/dataset_mapper.py b/grid_feats/dataset_mapper.py
@@ -189,8 +189,11 @@ def __init__(self, cfg, is_train=False):
 
     def __call__(self, dataset_dict):
         dataset_dict = copy.deepcopy(dataset_dict)
-        image = utils.read_image(dataset_dict["file_name"], format=self.img_format)
-        utils.check_image_size(dataset_dict, image)
+        try:
+            image = utils.read_image(dataset_dict["file_name"], format=self.img_format)
+            utils.check_image_size(dataset_dict, image)
+        except OSError:
+            return
 
         if "annotations" not in dataset_dict:
             image, transforms = T.apply_transform_gens(