From a2185a66341d382feb5948b000f4ddbbc3764c0e Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 8 Jun 2018 11:32:25 +0200 Subject: fix --- live-mogrify.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'live-mogrify.py') 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': -- cgit v1.2.3-70-g09d2