Modified the data_folder to be more generic
Compare changes
- Anjith GEORGE authored
@@ -12,9 +12,8 @@ import torch.utils.data as data
@@ -12,9 +12,8 @@ import torch.utils.data as data
@@ -102,7 +101,7 @@ class DataFolder(data.Dataset):
@@ -102,7 +101,7 @@ class DataFolder(data.Dataset):
@@ -155,7 +154,7 @@ class DataFolder(data.Dataset):
@@ -155,7 +154,7 @@ class DataFolder(data.Dataset):
@@ -238,9 +237,9 @@ class DataFolder(data.Dataset):
@@ -238,9 +237,9 @@ class DataFolder(data.Dataset):
@@ -254,20 +253,21 @@ class DataFolder(data.Dataset):
@@ -254,20 +253,21 @@ class DataFolder(data.Dataset):
if isinstance(self.transform, transforms.Compose): # if an instance of torchvision composed transformation
@@ -275,7 +275,7 @@ class DataFolder(data.Dataset):
@@ -275,7 +275,7 @@ class DataFolder(data.Dataset):
return torch.Tensor(img_array_transformed).unsqueeze(0), target # convert array to Tensor, also return target