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

check_img_size() update

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-06-16 17:48:01 -07:00
1 parent 034609414e
commit a557b7d924
1 file changed
+4 -3
Download Patch File Download Diff File Expand all files Collapse all files
utils/utils.py
+4 -3
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):