Browse Source

check_img_size() int() update

5.0
Glenn Jocher 4 years ago
parent
commit
4e62eb90be
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      utils/utils.py

+ 1
- 1
utils/utils.py View File

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

Loading…
Cancel
Save