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

Move task=classify with mode=track warning to tracker on_predict_start (#18837)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Laughing 2025-01-23 20:52:59 +08:00 • committed by GitHub
parent d6272f258e
commit d29023f85f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
ultralytics/cfg/__init__.py
View file

@ -921,12 +921,7 @@ def entrypoint(debug=""):
# Task # Task
task = overrides.pop("task", None) task = overrides.pop("task", None)
if task: if task:
if task == "classify" and mode == "track": if task not in TASKS:
raise ValueError(
f"❌ Classification doesn't support 'mode=track'. Valid modes for classification are"
f" {MODES - {'track'}}.\n{CLI_HELP_MSG}"
)
elif task not in TASKS:
if task == "track": if task == "track":
LOGGER.warning( LOGGER.warning(
"WARNING ⚠️ invalid 'task=track', setting 'task=detect' and 'mode=track'. Valid tasks are {TASKS}.\n{CLI_HELP_MSG}." "WARNING ⚠️ invalid 'task=track', setting 'task=detect' and 'mode=track'. Valid tasks are {TASKS}.\n{CLI_HELP_MSG}."

3
ultralytics/trackers/track.py
View file

@ -31,6 +31,9 @@ def on_predict_start(predictor: object, persist: bool = False) -> None:
>>> predictor = SomePredictorClass() >>> predictor = SomePredictorClass()
>>> on_predict_start(predictor, persist=True) >>> on_predict_start(predictor, persist=True)
""" """
if predictor.args.task == "classify":
raise ValueError("❌ Classification doesn't support 'mode=track'")
if hasattr(predictor, "trackers") and persist: if hasattr(predictor, "trackers") and persist:
return return