Merge cd00f0c into 6c047f0

Lightning-Universe · Feb 10, 2021 · d46b7d2 · d46b7d2
2 parents 6c047f0 + cd00f0c
commit d46b7d2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flash/vision/classification/data.py b/flash/vision/classification/data.py
@@ -68,6 +68,8 @@ def __len__(self) -> int:
     def __getitem__(self, index: int) -> Tuple[Any, Optional[int]]:
         filename = self.fnames[index]
         img = self.loader(filename)
+        if self.transform is not None:
+            img = self.transform(img)
         label = None
         if self.has_labels:
             label = self.labels[index]