glenn-jocher
commited on
Commit
•
4e62eb9
1
Parent(s):
2b7a113
check_img_size() int() update
Browse files- utils/utils.py +1 -1
utils/utils.py
CHANGED
@@ -47,7 +47,7 @@ def check_git_status():
|
|
47 |
|
48 |
def check_img_size(img_size, s=32):
|
49 |
# Verify img_size is a multiple of stride s
|
50 |
-
new_size = make_divisible(img_size, s) # ceil gs-multiple
|
51 |
if new_size != img_size:
|
52 |
print('WARNING: --img-size %g must be multiple of max stride %g, updating to %g' % (img_size, s, new_size))
|
53 |
return new_size
|
|
|
47 |
|
48 |
def check_img_size(img_size, s=32):
|
49 |
# Verify img_size is a multiple of stride s
|
50 |
+
new_size = make_divisible(img_size, int(s)) # ceil gs-multiple
|
51 |
if new_size != img_size:
|
52 |
print('WARNING: --img-size %g must be multiple of max stride %g, updating to %g' % (img_size, s, new_size))
|
53 |
return new_size
|