diff options
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': |