Standardize default region points (#17715)

pull/16600/head
Junyang Kong 7 days ago committed by GitHub
parent 3e48829cd6
commit 1b74f142a1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      tests/test_solutions.py
  2. 2
      ultralytics/cfg/solutions/default.yaml
  3. 2
      ultralytics/solutions/solutions.py

@ -16,7 +16,7 @@ def test_major_solutions():
safe_download(url=MAJOR_SOLUTIONS_DEMO) safe_download(url=MAJOR_SOLUTIONS_DEMO)
cap = cv2.VideoCapture("solutions_ci_demo.mp4") cap = cv2.VideoCapture("solutions_ci_demo.mp4")
assert cap.isOpened(), "Error reading video file" assert cap.isOpened(), "Error reading video file"
region_points = [(20, 400), (1080, 404), (1080, 360), (20, 360)] region_points = [(20, 400), (1080, 400), (1080, 360), (20, 360)]
counter = solutions.ObjectCounter(region=region_points, model="yolo11n.pt", show=False) # Test object counter counter = solutions.ObjectCounter(region=region_points, model="yolo11n.pt", show=False) # Test object counter
heatmap = solutions.Heatmap(colormap=cv2.COLORMAP_PARULA, model="yolo11n.pt", show=False) # Test heatmaps heatmap = solutions.Heatmap(colormap=cv2.COLORMAP_PARULA, model="yolo11n.pt", show=False) # Test heatmaps
speed = solutions.SpeedEstimator(region=region_points, model="yolo11n.pt", show=False) # Test queue manager speed = solutions.SpeedEstimator(region=region_points, model="yolo11n.pt", show=False) # Test queue manager

@ -2,7 +2,7 @@
# Configuration for Ultralytics Solutions # Configuration for Ultralytics Solutions
# Object counting settings # Object counting settings
region: # Object counting, queue or speed estimation region points. Default region points are [(20, 400), (1080, 404), (1080, 360), (20, 360)] region: # Object counting, queue or speed estimation region points. Default region points are [(20, 400), (1080, 400), (1080, 360), (20, 360)]
show_in: True # Flag to display objects moving *into* the defined region show_in: True # Flag to display objects moving *into* the defined region
show_out: True # Flag to display objects moving *out of* the defined region show_out: True # Flag to display objects moving *out of* the defined region

@ -135,7 +135,7 @@ class BaseSolution:
def initialize_region(self): def initialize_region(self):
"""Initialize the counting region and line segment based on configuration settings.""" """Initialize the counting region and line segment based on configuration settings."""
if self.region is None: if self.region is None:
self.region = [(20, 400), (1080, 404), (1080, 360), (20, 360)] self.region = [(20, 400), (1080, 400), (1080, 360), (20, 360)]
self.r_s = ( self.r_s = (
self.Polygon(self.region) if len(self.region) >= 3 else self.LineString(self.region) self.Polygon(self.region) if len(self.region) >= 3 else self.LineString(self.region)
) # region or line ) # region or line

Loading…
Cancel
Save