Apply transform in FilePathDataset (#97)

Co-authored-by: Teddy Koker <[email protected]> Co-authored-by: Kaushik B <[email protected]>
Lightning-Universe · Feb 10, 2021 · 7acd8b5 · 7acd8b5
1 parent 6c047f0
commit 7acd8b5
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]