Merge pull request #185 from delphieritas/patch-33

enable real image_size loading in ImageDataset4Clf
nusdbsystem · Mar 11, 2021 · 332ecdd · 332ecdd
2 parents 41d8bac + da673ae
commit 332ecdd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/singa_auto/datasets/image_classification_dataset.py b/singa_auto/datasets/image_classification_dataset.py
@@ -33,7 +33,8 @@ def __init__(self,
         self.min_image_size = min_image_size
         self.max_image_size = max_image_size
         self.label_mapper = dict()
-        self.image_size = None
+        # to read accual self.image_size
+        self.__getitem__(0)
         if if_shuffle:
             (self._image_names,
              self._image_classes) = self._shuffle(self._image_names,