Add return check for 'yolo solutions help' (#17518)

pull/17491/head^2
Muhammad Rizwan Munawar 2 weeks ago committed by GitHub
parent f9d889dd4d
commit 3682f01d12
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 3
      ultralytics/cfg/__init__.py
  2. 2
      ultralytics/solutions/analytics.py

@ -671,6 +671,9 @@ def handle_yolo_solutions(args: List[str]) -> None:
) )
s_n = "count" # Default solution if none provided 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 cls, method = SOLUTION_MAP[s_n] # solution class name, method name and default source
from ultralytics import solutions # import ultralytics solutions from ultralytics import solutions # import ultralytics solutions

@ -54,7 +54,7 @@ class Analytics(BaseSolution):
self.y_label = "Total Counts" self.y_label = "Total Counts"
# Predefined data # 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.fg_color = "#111E68" # foreground color of frame
self.title = "Ultralytics Solutions" # window name self.title = "Ultralytics Solutions" # window name
self.max_points = 45 # maximum points to be drawn on window self.max_points = 45 # maximum points to be drawn on window

Loading…
Cancel
Save