From 3682f01d123a3e0c2bbbd1a8504102f8cd56f384 Mon Sep 17 00:00:00 2001 From: Muhammad Rizwan Munawar Date: Wed, 13 Nov 2024 22:28:23 +0500 Subject: [PATCH] Add return check for 'yolo solutions help' (#17518) --- ultralytics/cfg/__init__.py | 3 +++ ultralytics/solutions/analytics.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/ultralytics/cfg/__init__.py b/ultralytics/cfg/__init__.py index c0675620b8..dff25f418c 100644 --- a/ultralytics/cfg/__init__.py +++ b/ultralytics/cfg/__init__.py @@ -671,6 +671,9 @@ def handle_yolo_solutions(args: List[str]) -> None: ) s_n = "count" # Default solution if none provided + if args and args[0] == "help": # Add check for return if user call `yolo solutions help` + return + cls, method = SOLUTION_MAP[s_n] # solution class name, method name and default source from ultralytics import solutions # import ultralytics solutions diff --git a/ultralytics/solutions/analytics.py b/ultralytics/solutions/analytics.py index aed7beed94..9be192448c 100644 --- a/ultralytics/solutions/analytics.py +++ b/ultralytics/solutions/analytics.py @@ -54,7 +54,7 @@ class Analytics(BaseSolution): self.y_label = "Total Counts" # Predefined data - self.bg_color = "#00F344" # background color of frame + self.bg_color = "#F3F3F3" # background color of frame self.fg_color = "#111E68" # foreground color of frame self.title = "Ultralytics Solutions" # window name self.max_points = 45 # maximum points to be drawn on window