From 2aa96a6bf070804c7a3231b6a93939c02527dfc7 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 8 Jun 2018 11:30:53 +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 6817fd7..35fe797 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -160,9 +160,9 @@ import torchvision.transforms as transforms def get_transform(opt={}): transform_list = [] - # if opt.resize_or_crop == 'resize_and_crop': - # osize = [opt.loadSize, opt.loadSize] - # transform_list.append(transforms.Scale(osize, Image.BICUBIC)) + 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: @@ -190,7 +190,7 @@ def load_frame(opt, index): if not os.path.exists(A_path): print("path doesn't exist: {}".format(A_path)) return None - transform = get_transform() + transform = get_transform(opt) A_img = Image.open(A_path).convert('RGB') A = transform(A_img) # if self.opt.which_direction == 'BtoA': -- cgit v1.2.3-70-g09d2