diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-05-12 18:03:02 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-05-12 18:03:02 +0200 |
| commit | 6e02d55fa01b0532ac9f152765b815a0415310b2 (patch) | |
| tree | dc79a35368e94e8c3f5ac72c475a542bc5b8985b /data/base_dataset.py | |
| parent | 0ad66a09aaf55023d8eaff70094b0798b01adb19 (diff) | |
| parent | 0e03791f3f7db3f157bed785ba06d21270b98059 (diff) | |
merge
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': |