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

Rename img_size to imgsz (#86)

Browse source
This commit is contained in:
Glenn Jocher 2022-12-24 00:39:09 +01:00 • committed by GitHub
parent ae2443c210
commit 6432afc5f9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
25 changed files with 98 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ultralytics/yolo/engine/model.py
View file

@ -111,11 +111,11 @@ class YOLO:
predictor = self.PredictorClass(overrides=kwargs)
# check size type
sz = predictor.args.img_size
sz = predictor.args.imgsz
if type(sz) != int: # recieved listConfig
predictor.args.img_size = [sz[0], sz[0]] if len(sz) == 1 else [sz[0], sz[1]] # expand
predictor.args.imgsz = [sz[0], sz[0]] if len(sz) == 1 else [sz[0], sz[1]] # expand
else:
predictor.args.img_size = [sz, sz]
predictor.args.imgsz = [sz, sz]
predictor.setup(model=self.model, source=source)
predictor()