update aligned_dataset

1 files changed, 1 insertions, 9 deletions

diff --git a/data/aligned_dataset.py b/data/aligned_dataset.py
index 08155c8..8899cb2 100644
--- a/data/aligned_dataset.py
+++ b/data/aligned_dataset.py
@@ -12,17 +12,9 @@ class AlignedDataset(BaseDataset):
         self.opt = opt
         self.root = opt.dataroot
         self.dir_AB = os.path.join(opt.dataroot, opt.phase)
-
         self.AB_paths = sorted(make_dataset(self.dir_AB))
-
         assert(opt.resize_or_crop == 'resize_and_crop')
 
-        transform_list = [transforms.ToTensor(),
-                          transforms.Normalize((0.5, 0.5, 0.5),
-                                               (0.5, 0.5, 0.5))]
-
-        self.transform = transforms.Compose(transform_list)
-
     def __getitem__(self, index):
         AB_path = self.AB_paths[index]
         AB = Image.open(AB_path).convert('RGB')
@@ -63,7 +55,7 @@ class AlignedDataset(BaseDataset):
         if output_nc == 1:  # RGB to gray
             tmp = B[0, ...] * 0.299 + B[1, ...] * 0.587 + B[2, ...] * 0.114
             B = tmp.unsqueeze(0)
-    
+
         return {'A': A, 'B': B,
                 'A_paths': AB_path, 'B_paths': AB_path}