From 9a89186b52ceaa3b99292ebac0b71eb992631259 Mon Sep 17 00:00:00 2001 From: Muhammad Rizwan Munawar Date: Mon, 13 Jan 2025 15:53:42 +0500 Subject: [PATCH] Add warning for `task=classify` with `mode=track` (#18621) Co-authored-by: UltralyticsAssistant Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com> --- ultralytics/cfg/__init__.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ultralytics/cfg/__init__.py b/ultralytics/cfg/__init__.py index cc15b135..bd06af1d 100644 --- a/ultralytics/cfg/__init__.py +++ b/ultralytics/cfg/__init__.py @@ -913,7 +913,12 @@ def entrypoint(debug=""): # Task task = overrides.pop("task", None) if task: - if task not in TASKS: + if task == "classify" and mode == "track": + 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: raise ValueError(f"Invalid 'task={task}'. Valid tasks are {TASKS}.\n{CLI_HELP_MSG}") if "model" not in overrides: overrides["model"] = TASK2MODEL[task]