From 5bf446e4007f9b1c657c32f817e78437afb35c04 Mon Sep 17 00:00:00 2001 From: junyanz Date: Sat, 22 Apr 2017 21:39:34 -0700 Subject: fix cropping in data_loaders --- data/aligned_data_loader.py | 4 +--- data/unaligned_data_loader.py | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) (limited to 'data') diff --git a/data/aligned_data_loader.py b/data/aligned_data_loader.py index c5305fc..8171bc2 100644 --- a/data/aligned_data_loader.py +++ b/data/aligned_data_loader.py @@ -25,7 +25,6 @@ class PairedData(object): w_offset = random.randint(0, max(0, w - self.fineSize - 1)) h_offset = random.randint(0, max(0, h - self.fineSize - 1)) - A = AB[:, :, h_offset:h_offset + self.fineSize, w_offset:w_offset + self.fineSize] B = AB[:, :, h_offset:h_offset + self.fineSize, @@ -40,8 +39,7 @@ class AlignedDataLoader(BaseDataLoader): self.fineSize = opt.fineSize transform = transforms.Compose([ # TODO: Scale - #transforms.Scale((opt.loadSize * 2, opt.loadSize)), - #transforms.CenterCrop(opt.fineSize), + transforms.Scale(opt.loadSize), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]) diff --git a/data/unaligned_data_loader.py b/data/unaligned_data_loader.py index f37f6ce..6e926d3 100644 --- a/data/unaligned_data_loader.py +++ b/data/unaligned_data_loader.py @@ -52,7 +52,7 @@ class UnalignedDataLoader(BaseDataLoader): BaseDataLoader.initialize(self, opt) transform = transforms.Compose([ transforms.Scale(opt.loadSize), - transforms.CenterCrop(opt.fineSize), + transforms.RandomCrop(opt.fineSize), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]) -- cgit v1.2.3-70-g09d2