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

Add train.py ``--img-size` floor (#4099)

This commit is contained in:
Glenn Jocher 2021-07-21 16:50:47 +02:00 committed by GitHub
parent 3bef77f5cb
commit 2c073cd207
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -207,7 +207,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
# Image sizes # Image sizes
gs = max(int(model.stride.max()), 32) # grid size (max stride) gs = max(int(model.stride.max()), 32) # grid size (max stride)
nl = model.model[-1].nl # number of detection layers (used for scaling hyp['obj']) nl = model.model[-1].nl # number of detection layers (used for scaling hyp['obj'])
imgsz = check_img_size(opt.imgsz, gs) # verify imgsz is gs-multiple imgsz = check_img_size(opt.imgsz, gs, floor=gs * 2) # verify imgsz is gs-multiple
# DP mode # DP mode
if cuda and RANK == -1 and torch.cuda.device_count() > 1: if cuda and RANK == -1 and torch.cuda.device_count() > 1:

6
utils/general.py
View File

@ -181,11 +181,11 @@ def check_requirements(requirements='requirements.txt', exclude=()):
print(emojis(s)) # emoji-safe print(emojis(s)) # emoji-safe
def check_img_size(img_size, s=32): def check_img_size(img_size, s=32, floor=0):
# Verify img_size is a multiple of stride s # Verify img_size is a multiple of stride s
new_size = make_divisible(img_size, int(s)) # ceil gs-multiple new_size = max(make_divisible(img_size, int(s)), floor) # 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(f'WARNING: --img-size {img_size} must be multiple of max stride {s}, updating to {new_size}')
return new_size return new_size