diff options
| author | jules@lens <julescarbon@gmail.com> | 2018-05-08 21:30:57 +0200 |
|---|---|---|
| committer | jules@lens <julescarbon@gmail.com> | 2018-05-08 21:30:57 +0200 |
| commit | 2f4375e03ba13625497a5c4bae381a6ad0c6f937 (patch) | |
| tree | 5126460a7eb13d06493038b7a6be9a8d4a797827 /data/base_dataset.py | |
| parent | 5979723a35c577ffe70c1107415199444fb5c7fe (diff) | |
new dataset stuff
Diffstat (limited to 'data/base_dataset.py')
| -rw-r--r-- | data/base_dataset.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/data/base_dataset.py b/data/base_dataset.py index 7cfac54..12fd0b2 100644 --- a/data/base_dataset.py +++ b/data/base_dataset.py @@ -19,7 +19,10 @@ def get_transform(opt): if opt.resize_or_crop == 'resize_and_crop': osize = [opt.loadSize, opt.loadSize] transform_list.append(transforms.Scale(osize, Image.BICUBIC)) - transform_list.append(transforms.RandomCrop(opt.fineSize)) + if opt.center_crop: + transform_list.append(transforms.CenterCrop(opt.fineSize)) + else: + transform_list.append(transforms.RandomCrop(opt.fineSize)) elif opt.resize_or_crop == 'crop': transform_list.append(transforms.RandomCrop(opt.fineSize)) elif opt.resize_or_crop == 'scale_width': |