Upload utils.py
Browse files- libs/utils.py +1 -1
libs/utils.py
CHANGED
|
@@ -29,7 +29,7 @@ def resize(img, resize=512):
|
|
| 29 |
scale = resize/h
|
| 30 |
else:
|
| 31 |
scale = resize/w
|
| 32 |
-
return cv2.resize(img, (w * scale, h * scale))
|
| 33 |
|
| 34 |
def to_tensor(img):
|
| 35 |
r"""
|
|
|
|
| 29 |
scale = resize/h
|
| 30 |
else:
|
| 31 |
scale = resize/w
|
| 32 |
+
return cv2.resize(img, (int(w * scale), int(h * scale)))
|
| 33 |
|
| 34 |
def to_tensor(img):
|
| 35 |
r"""
|