Merge pull request #30 from ppjerry/FixDataTransformation

fix the random transformation on both image and target

Author: bigmb

Data_Loader.py

@@ -6,6 +6,8 @@
 import torchvision
 from skimage import io
 from torch.utils.data import Dataset
+import random
+import numpy as np
 
 
 class Images_Dataset(Dataset):
@@ -80,6 +82,7 @@ def __init__(self, images_dir, labels_dir,transformI = None, transformM = None):
  self.lx = torchvision.transforms.Compose([
  # torchvision.transforms.Resize((128,128)),
  torchvision.transforms.CenterCrop(96),
+ torchvision.transforms.RandomRotation((-10,10)),
  torchvision.transforms.Grayscale(),
  torchvision.transforms.ToTensor(),
  #torchvision.transforms.Lambda(lambda x: torch.cat([x, 1 - x], dim=0))
@@ -93,5 +96,19 @@ def __getitem__(self, i):
  i1 = Image.open(self.images_dir + self.images[i])
  l1 = Image.open(self.labels_dir + self.labels[i])
 
- return self.tx(i1), self.lx(l1)
+ seed=np.random.randint(0,2**32) # make a seed with numpy generator 
+
+ # apply this seed to img tranfsorms
+ random.seed(seed) 
+ torch.manual_seed(seed)
+ img = self.tx(i1)
+ 
+ # apply this seed to target/label tranfsorms 
+ random.seed(seed) 
+ torch.manual_seed(seed)
+ label = self.lx(l1)
+
+ 
+
+ return img, label