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

ultralytics 8.2.57 new Solutions Tests and Docs (#14408)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Muhammad Rizwan Munawar 2024-07-14 23:00:14 +05:00 • committed by GitHub
parent 100a73b6e9
commit c67a3039c1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
12 changed files with 143 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

8
ultralytics/solutions/queue_management.py
View file

@ -17,7 +17,7 @@ class QueueManager:
def __init__(
self,
classes_names,
names,
reg_pts=None,
line_thickness=2,
track_thickness=2,
@ -34,7 +34,7 @@ class QueueManager:
Initializes the QueueManager with specified parameters for tracking and counting objects.
Args:
classes_names (dict): A dictionary mapping class IDs to class names.
names (dict): A dictionary mapping class IDs to class names.
reg_pts (list of tuples, optional): Points defining the counting region polygon. Defaults to a predefined
rectangle.
line_thickness (int, optional): Thickness of the annotation lines. Defaults to 2.
@ -69,7 +69,7 @@ class QueueManager:
self.view_queue_counts = view_queue_counts
self.fontsize = fontsize
self.names = classes_names # Class names
self.names = names # Class names
self.annotator = None # Annotator
self.window_name = "Ultralytics YOLOv8 Queue Manager"
@ -139,7 +139,7 @@ class QueueManager:
def display_frames(self):
"""Displays the current frame with annotations."""
if self.env_check:
if self.env_check and self.view_img:
self.annotator.draw_region(reg_pts=self.reg_pts, thickness=self.region_thickness, color=self.region_color)
cv2.namedWindow(self.window_name)
cv2.imshow(self.window_name, self.im0)