提交 4e62eb90 authored 作者: Glenn Jocher's avatar Glenn Jocher

check_img_size() int() update

上级 2b7a1130
...@@ -47,7 +47,7 @@ def check_git_status(): ...@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论