summaryrefslogtreecommitdiff
path: root/data/unaligned_data_loader.py
diff options
context:
space:
mode:
authorjunyanz <junyanz@berkeley.edu>2017-04-22 21:39:34 -0700
committerjunyanz <junyanz@berkeley.edu>2017-04-22 21:39:34 -0700
commit5bf446e4007f9b1c657c32f817e78437afb35c04 (patch)
treeb8a4d5caf40db4aecaf1b048adc0971a613b8b6a /data/unaligned_data_loader.py
parent9137b1146f9b0edb14d7c4421ed7abed1583b025 (diff)
fix cropping in data_loaders
Diffstat (limited to 'data/unaligned_data_loader.py')
-rw-r--r--data/unaligned_data_loader.py2
1 files changed, 1 insertions, 1 deletions
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))])