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 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': |