check_img_size() int() update
This commit is contained in:
1 parent
2b7a11300e
commit
4e62eb90be
1 file changed
+1
-1
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user