From 2f4375e03ba13625497a5c4bae381a6ad0c6f937 Mon Sep 17 00:00:00 2001 From: "jules@lens" Date: Tue, 8 May 2018 21:30:57 +0200 Subject: new dataset stuff --- data/base_dataset.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'data/base_dataset.py') 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': -- cgit v1.2.3-70-g09d2