From 2c073cd207bae1163b472c561d3fd31b1d2ba870 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 21 Jul 2021 16:50:47 +0200 Subject: [PATCH] Add train.py ``--img-size` floor (#4099) --- train.py | 2 +- utils/general.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/train.py b/train.py index b1afaf8ada75..9a844ebac0de 100644 --- a/train.py +++ b/train.py @@ -207,7 +207,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary # Image sizes 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']) - 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 if cuda and RANK == -1 and torch.cuda.device_count() > 1: diff --git a/utils/general.py b/utils/general.py index 08a3ff6539b2..fabd0f35fe9e 100755 --- a/utils/general.py +++ b/utils/general.py @@ -181,11 +181,11 @@ def check_requirements(requirements='requirements.txt', exclude=()): 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 - 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: - 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