From beed9ba76a88da7bb4fe09094fe6e175f6b2d0de Mon Sep 17 00:00:00 2001 From: junyanz Date: Sun, 14 Jan 2018 17:19:29 -0800 Subject: update aligned_dataset --- data/aligned_dataset.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) (limited to 'data/aligned_dataset.py') 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} -- cgit v1.2.3-70-g09d2