vobecant
commited on
Commit
·
c6898bc
1
Parent(s):
81ad5a8
Initial commit.
Browse files
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 |
-
|
|
|
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 |
|