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