Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

check_img_size() int() update

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-06-30 16:41:26 -07:00
1 parent 2b7a11300e
commit 4e62eb90be
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
utils/utils.py
+1 -1
View File
@@ -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