HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Cleanup (#168)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kalen Michael <kalenmike@gmail.com>
This commit is contained in:
Glenn Jocher 2023-01-09 23:22:33 +01:00 • committed by GitHub
parent 6feba17760
commit 422c49d439
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
97 changed files with 224 additions and 757 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ultralytics/yolo/utils/checks.py
View file

@ -1,3 +1,5 @@
# Ultralytics YOLO 🚀, GPL-3.0 license
import glob
import inspect
import math
@ -62,8 +64,7 @@ def check_imgsz(imgsz, stride=32, min_dim=1, floor=0):
LOGGER.warning(f'WARNING ⚠️ --img-size {imgsz} must be multiple of max stride {stride}, updating to {sz}')
# Add missing dimensions if necessary
if min_dim == 2 and len(sz) == 1:
sz = [sz[0], sz[0]]
sz = [sz[0], sz[0]] if min_dim == 2 and len(sz) == 1 else sz[0] if min_dim == 1 and len(sz) == 1 else sz
return sz