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

Model enhancements 2 (#109)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Ayush Chaurasia 2022-12-28 21:01:51 +05:30 • committed by GitHub
parent 8406b49b49
commit 0303ced8ab
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 14 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -112,8 +112,9 @@ class YOLO:
source (str): Accepts all source types accepted by yolo
**kwargs : Any other args accepted by the predictors. Too see all args check 'configuration' section in the docs
"""
kwargs.update(self.overrides)
predictor = self.PredictorClass(overrides=kwargs)
overrides = self.overrides.copy()
overrides.update(kwargs)
predictor = self.PredictorClass(overrides=overrides)
# check size type
sz = predictor.args.imgsz
@ -136,8 +137,9 @@ class YOLO:
if not self.model:
raise Exception("model not initialized!")
kwargs.update(self.overrides)
args = get_config(config=DEFAULT_CONFIG, overrides=kwargs)
overrides = self.overrides.copy()
overrides.update(kwargs)
args = get_config(config=DEFAULT_CONFIG, overrides=overrides)
args.data = data or args.data
args.task = self.task