IgnoreLee commited on
Commit
669e9b3
·
1 Parent(s): 67c3f29

Upload utils.py

Browse files
Files changed (1) hide show
  1. 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"""