diff --git a/utils/utils.py b/utils/utils.py index c33f41f..305486a 100755 --- 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