zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

check_img_size() update

This commit is contained in:
Glenn Jocher 2020-06-16 17:48:01 -07:00
parent 034609414e
commit a557b7d924
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
utils/utils.py
View File

@ -47,9 +47,10 @@ 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
if img_size % s != 0: new_size = make_divisible(img_size, s) # ceil gs-multiple
print('WARNING: --img-size %g must be multiple of max stride %g' % (img_size, s)) if new_size != img_size:
return make_divisible(img_size, s) # nearest gs-multiple print('WARNING: --img-size %g must be multiple of max stride %g, updating to %g' % (img_size, s, new_size))
return new_size
def check_anchors(dataset, model, thr=4.0, imgsz=640): def check_anchors(dataset, model, thr=4.0, imgsz=640):