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

Integrate ByteTracker and BoT-SORT trackers (#788)

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>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
This commit is contained in:
Laughing 2023-02-16 00:23:03 +08:00 • committed by GitHub
parent d99e04daa1
commit ed6c54da7a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
24 changed files with 1635 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ultralytics/yolo/cfg/__init__.py
View file

@ -197,9 +197,7 @@ def entrypoint(debug=''):
# Define tasks and modes
tasks = 'detect', 'segment', 'classify'
modes = 'train', 'val', 'predict', 'export'
# Define special commands
modes = 'train', 'val', 'predict', 'export', 'track'
special = {
'help': lambda: LOGGER.info(CLI_HELP_MSG),
'checks': checks.check_yolo,
@ -288,7 +286,7 @@ def entrypoint(debug=''):
f"Inheriting 'task={model.task}' from {overrides['model']} and ignoring 'task={task}'.")
task = model.task
overrides['task'] = task
if mode == 'predict' and 'source' not in overrides:
if mode in {'predict', 'track'} and 'source' not in overrides:
overrides['source'] = DEFAULT_CFG.source or ROOT / "assets" if (ROOT / "assets").exists() \
else "https://ultralytics.com/images/bus.jpg"
LOGGER.warning(f"WARNING ⚠️ 'source' is missing. Using default 'source={overrides['source']}'.")