diff options
| -rw-r--r-- | live-mogrify.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index 0725753..f4200a3 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -158,7 +158,7 @@ def list_sequences(module): import torchvision.transforms as transforms -def get_transform(opt): +def get_transform(opt={}): transform_list = [] # if opt.resize_or_crop == 'resize_and_crop': # osize = [opt.loadSize, opt.loadSize] @@ -190,8 +190,9 @@ 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() A = Image.open(A_path).convert('RGB') - A = self.transform(A_img) + A = transform(A_img) # if self.opt.which_direction == 'BtoA': # input_nc = self.opt.output_nc # else: |