diff options
| -rw-r--r-- | data/aligned_dataset.py | 10 |
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} |