diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-22 21:39:34 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-22 21:39:34 -0700 |
| commit | 5bf446e4007f9b1c657c32f817e78437afb35c04 (patch) | |
| tree | b8a4d5caf40db4aecaf1b048adc0971a613b8b6a /data | |
| parent | 9137b1146f9b0edb14d7c4421ed7abed1583b025 (diff) | |
fix cropping in data_loaders
Diffstat (limited to 'data')
| -rw-r--r-- | data/aligned_data_loader.py | 4 | ||||
| -rw-r--r-- | data/unaligned_data_loader.py | 2 |
2 files changed, 2 insertions, 4 deletions
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))]) |