vobecant commited on
Commit
c6898bc
·
1 Parent(s): 81ad5a8

Initial commit.

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -141,8 +141,9 @@ def get_transformations(input_img):
141
 
142
  shorter_input_size = min(input_img.size)
143
 
144
- if im_size != 1024 or shorter_input_size < im_size:
145
- trans_list.append(transforms.Resize(im_size))
 
146
 
147
  trans_list.append(transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]))
148
 
 
141
 
142
  shorter_input_size = min(input_img.size)
143
 
144
+ # if im_size != 1024 or shorter_input_size < im_size:
145
+ # trans_list.append(transforms.Resize(im_size))
146
+ trans_list.append(transforms.Resize(im_size))
147
 
148
  trans_list.append(transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]))
149