JessicaDuFirst · October 24, 2019 13:10 · Jul 4, 2019 · Nov 5, 2018 · Jun 27, 2018 · Jun 27, 2018
diff --git a/pytorch_image_folder_with_file_paths.py b/pytorch_image_folder_with_file_paths.py
@@ -6,7 +6,7 @@ class ImageFolderWithPaths(datasets.ImageFolder):
     torchvision.datasets.ImageFolder
     """
 
-    # override the __getitem__ method. this is the method dataloader calls
+    # override the __getitem__ method. this is the method that dataloader calls
     def __getitem__(self, index):
         # this is what ImageFolder normally returns 
         original_tuple = super(ImageFolderWithPaths, self).__getitem__(index)

diff --git a/pytorch_image_folder_with_file_paths.py b/pytorch_image_folder_with_file_paths.py
@@ -15,7 +15,8 @@ def __getitem__(self, index):
         # make a new tuple that includes original and the path
         tuple_with_path = (original_tuple + (path,))
         return tuple_with_path
-
+
+# EXAMPLE USAGE:
 # instantiate the dataset and dataloader
 data_dir = "your/data_dir/here"
 dataset = ImageFolderWithPaths(data_dir) # our custom dataset

diff --git a/pytorch_image_folder_with_file_paths.py b/pytorch_image_folder_with_file_paths.py
@@ -2,7 +2,7 @@
 from torchvision import datasets
 
 class ImageFolderWithPaths(datasets.ImageFolder):
-    """Custom dataset that includes image paths. Extends
+    """Custom dataset that includes image file paths. Extends
     torchvision.datasets.ImageFolder
     """
 

diff --git a/pytorch_image_folder_with_file_paths.py b/pytorch_image_folder_with_file_paths.py
@@ -6,7 +6,7 @@ class ImageFolderWithPaths(datasets.ImageFolder):
     torchvision.datasets.ImageFolder
     """
 
-    # override the __getitem__ method that dataloader calls
+    # override the __getitem__ method. this is the method dataloader calls
     def __getitem__(self, index):
         # this is what ImageFolder normally returns 
         original_tuple = super(ImageFolderWithPaths, self).__getitem__(index)

diff --git a/pytorch_image_folder_with_file_paths.py b/pytorch_image_folder_with_file_paths.py
@@ -24,4 +24,4 @@ def __getitem__(self, index):
 # iterate over data
 for inputs, labels, paths in dataloader:
     # use the above variables freely
-    pass
+    print(inputs, labels, paths)
diff --git a/pytorch_image_folder_with_file_paths.py b/pytorch_image_folder_with_file_paths.py
@@ -21,7 +21,7 @@ def __getitem__(self, index):
 dataset = ImageFolderWithPaths(data_dir) # our custom dataset
 dataloader = torch.utils.DataLoader(dataset)
 
-# iterate over data like this
+# iterate over data
 for inputs, labels, paths in dataloader:
     # use the above variables freely
     pass
diff --git a/pytorch_image_folder_with_file_paths.py b/pytorch_image_folder_with_file_paths.py
@@ -18,9 +18,10 @@ def __getitem__(self, index):
 
 # instantiate the dataset and dataloader
 data_dir = "your/data_dir/here"
-dataset = ImageFolderWithPaths(data_dir)
+dataset = ImageFolderWithPaths(data_dir) # our custom dataset
 dataloader = torch.utils.DataLoader(dataset)
 
+# iterate over data like this
 for inputs, labels, paths in dataloader:
     # use the above variables freely
     pass
diff --git a/pytorch_image_folder_with_file_paths.py b/pytorch_image_folder_with_file_paths.py
@@ -0,0 +1,26 @@
+import torch
+from torchvision import datasets
+
+class ImageFolderWithPaths(datasets.ImageFolder):
+    """Custom dataset that includes image paths. Extends
+    torchvision.datasets.ImageFolder
+    """
+
+    # override the __getitem__ method that dataloader calls
+    def __getitem__(self, index):
+        # this is what ImageFolder normally returns 
+        original_tuple = super(ImageFolderWithPaths, self).__getitem__(index)
+        # the image file path
+        path = self.imgs[index][0]
+        # make a new tuple that includes original and the path
+        tuple_with_path = (original_tuple + (path,))
+        return tuple_with_path
+
+# instantiate the dataset and dataloader
+data_dir = "your/data_dir/here"
+dataset = ImageFolderWithPaths(data_dir)
+dataloader = torch.utils.DataLoader(dataset)
+
+for inputs, labels, paths in dataloader:
+    # use the above variables freely
+    pass