check_img_size() int() update

pull/1/head
Glenn Jocher 5 years ago
parent 2b7a11300e
commit 4e62eb90be

@ -47,7 +47,7 @@ def check_git_status():
def check_img_size(img_size, s=32): def check_img_size(img_size, s=32):
# Verify img_size is a multiple of stride s # 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: 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)) print('WARNING: --img-size %g must be multiple of max stride %g, updating to %g' % (img_size, s, new_size))
return new_size return new_size

Loading…
Cancel
Save