diff options
| -rw-r--r-- | live-mogrify.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index 35fe797..d93991b 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -163,10 +163,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)) - # if opt.center_crop: - # transform_list.append(transforms.CenterCrop(opt.fineSize)) - # else: - # 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': |