check_img_size() int() update

5 years ago · 4e62eb90be
parent 2b7a11300e
commit 4e62eb90be
1 changed files with 1 additions and 1 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -47,7 +47,7 @@ def check_git_status():

 def check_img_size(img_size, s=32):
    # Verify img_size is a multiple of stride s
-    new_size = make_divisible(img_size, s)  # ceil gs-multiple
+    new_size = make_divisible(img_size, int(s))  # ceil gs-multiple
    if new_size != img_size:
        print('WARNING: --img-size %g must be multiple of max stride %g, updating to %g' % (img_size, s, new_size))
    return new_size