New 🌟 Queue-Management code & docs (#9494)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Muhammad Rizwan Munawar
GitHub
parent
b3ac2f3951
commit
de2641fad7
5 changed files with 374 additions and 0 deletions
|
|
@ -56,6 +56,7 @@ Here's a compilation of in-depth guides to help you master different aspects of
|
|||
- [VisionEye View Objects Mapping](vision-eye.md) 🚀 NEW: This feature aim computers to discern and focus on specific objects, much like the way the human eye observes details from a particular viewpoint.
|
||||
- [Speed Estimation](speed-estimation.md) 🚀 NEW: Speed estimation in computer vision relies on analyzing object motion through techniques like [object tracking](https://docs.ultralytics.com/modes/track/), crucial for applications like autonomous vehicles and traffic monitoring.
|
||||
- [Distance Calculation](distance-calculation.md) 🚀 NEW: Distance calculation, which involves measuring the separation between two objects within a defined space, is a crucial aspect. In the context of Ultralytics YOLOv8, the method employed for this involves using the bounding box centroid to determine the distance associated with user-highlighted bounding boxes.
|
||||
- [Queue Management](queue-management.md) 🚀 NEW: Queue management is the practice of efficiently controlling and directing the flow of people or tasks, often through strategic planning and technology implementation, to minimize wait times and improve overall productivity.
|
||||
|
||||
## Contribute to Our Guides
|
||||
|
||||
|
|
|
|||
154
docs/en/guides/queue-management.md
Normal file
154
docs/en/guides/queue-management.md
Normal file
|
|
@ -0,0 +1,154 @@
|
|||
---
|
||||
comments: true
|
||||
description: Queue Management Using Ultralytics YOLOv8
|
||||
keywords: Ultralytics, YOLOv8, Queue Management, Object Counting, Object Tracking, Object Detection, Notebook, IPython Kernel, CLI, Python SDK
|
||||
---
|
||||
|
||||
# Queue Management using Ultralytics YOLOv8 🚀
|
||||
|
||||
## What is Queue Management?
|
||||
|
||||
Queue management using [Ultralytics YOLOv8](https://github.com/ultralytics/ultralytics/) involves organizing and controlling lines of people or vehicles to reduce wait times and enhance efficiency. It's about optimizing queues to improve customer satisfaction and system performance in various settings like retail, banks, airports, and healthcare facilities.
|
||||
|
||||
|
||||
## Advantages of Queue Management?
|
||||
|
||||
- **Reduced Waiting Times:** Queue management systems efficiently organize queues, minimizing wait times for customers. This leads to improved satisfaction levels as customers spend less time waiting and more time engaging with products or services.
|
||||
- **Increased Efficiency:** Implementing queue management allows businesses to allocate resources more effectively. By analyzing queue data and optimizing staff deployment, businesses can streamline operations, reduce costs, and improve overall productivity.
|
||||
|
||||
## Real World Applications
|
||||
|
||||
| Logistics | Retail |
|
||||
|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|:----------------------------------------------------------------------------------------------------------------------------------------------------------:|
|
||||
|  |  |
|
||||
| Queue management at airport ticket counter Using Ultralytics YOLOv8 | Queue monitoring in crowd Ultralytics YOLOv8 |
|
||||
|
||||
|
||||
!!! Example "Queue Management using YOLOv8 Example"
|
||||
|
||||
=== "Queue Manager"
|
||||
|
||||
```python
|
||||
import cv2
|
||||
from ultralytics import YOLO
|
||||
from ultralytics.solutions import queue_management
|
||||
|
||||
model = YOLO("yolov8n.pt")
|
||||
cap = cv2.VideoCapture("path/to/video/file.mp4")
|
||||
|
||||
assert cap.isOpened(), "Error reading video file"
|
||||
w, h, fps = (int(cap.get(x)) for x in (cv2.CAP_PROP_FRAME_WIDTH,
|
||||
cv2.CAP_PROP_FRAME_HEIGHT,
|
||||
cv2.CAP_PROP_FPS))
|
||||
|
||||
video_writer = cv2.VideoWriter("queue_management.avi",
|
||||
cv2.VideoWriter_fourcc(*'mp4v'),
|
||||
fps,
|
||||
(w, h))
|
||||
|
||||
queue_region = [(20, 400), (1080, 404), (1080, 360), (20, 360)]
|
||||
|
||||
queue = queue_management.QueueManager()
|
||||
queue.set_args(classes_names=model.names,
|
||||
reg_pts=queue_region,
|
||||
line_thickness=3,
|
||||
fontsize=1.0,
|
||||
region_color=(255, 144, 31))
|
||||
|
||||
while cap.isOpened():
|
||||
success, im0 = cap.read()
|
||||
|
||||
if success:
|
||||
tracks = model.track(im0, show=False, persist=True,
|
||||
verbose=False)
|
||||
out = queue.process_queue(im0, tracks)
|
||||
|
||||
video_writer.write(im0)
|
||||
if cv2.waitKey(1) & 0xFF == ord('q'):
|
||||
break
|
||||
continue
|
||||
|
||||
print("Video frame is empty or video processing has been successfully completed.")
|
||||
break
|
||||
|
||||
cap.release()
|
||||
cv2.destroyAllWindows()
|
||||
```
|
||||
|
||||
=== "Queue Manager Specific Classes"
|
||||
|
||||
```python
|
||||
import cv2
|
||||
from ultralytics import YOLO
|
||||
from ultralytics.solutions import queue_management
|
||||
|
||||
model = YOLO("yolov8n.pt")
|
||||
cap = cv2.VideoCapture("path/to/video/file.mp4")
|
||||
|
||||
assert cap.isOpened(), "Error reading video file"
|
||||
w, h, fps = (int(cap.get(x)) for x in (cv2.CAP_PROP_FRAME_WIDTH,
|
||||
cv2.CAP_PROP_FRAME_HEIGHT,
|
||||
cv2.CAP_PROP_FPS))
|
||||
|
||||
video_writer = cv2.VideoWriter("queue_management.avi",
|
||||
cv2.VideoWriter_fourcc(*'mp4v'),
|
||||
fps,
|
||||
(w, h))
|
||||
|
||||
queue_region = [(20, 400), (1080, 404), (1080, 360), (20, 360)]
|
||||
|
||||
queue = queue_management.QueueManager()
|
||||
queue.set_args(classes_names=model.names,
|
||||
reg_pts=queue_region,
|
||||
line_thickness=3,
|
||||
fontsize=1.0,
|
||||
region_color=(255, 144, 31))
|
||||
|
||||
while cap.isOpened():
|
||||
success, im0 = cap.read()
|
||||
|
||||
if success:
|
||||
tracks = model.track(im0, show=False, persist=True,
|
||||
verbose=False, classes=0) # Only person class
|
||||
out = queue.process_queue(im0, tracks)
|
||||
|
||||
video_writer.write(im0)
|
||||
if cv2.waitKey(1) & 0xFF == ord('q'):
|
||||
break
|
||||
continue
|
||||
|
||||
print("Video frame is empty or video processing has been successfully completed.")
|
||||
break
|
||||
|
||||
cap.release()
|
||||
cv2.destroyAllWindows()
|
||||
```
|
||||
|
||||
### Optional Arguments `set_args`
|
||||
|
||||
| Name | Type | Default | Description |
|
||||
|-----------------------|-------------|----------------------------|---------------------------------------------|
|
||||
| `view_img` | `bool` | `False` | Display frames with counts |
|
||||
| `view_queue_counts` | `bool` | `True` | Display Queue counts only on video frame |
|
||||
| `line_thickness` | `int` | `2` | Increase bounding boxes thickness |
|
||||
| `reg_pts` | `list` | `[(20, 400), (1260, 400)]` | Points defining the Region Area |
|
||||
| `classes_names` | `dict` | `model.model.names` | Dictionary of Class Names |
|
||||
| `region_color` | `RGB Color` | `(255, 0, 255)` | Color of the Object counting Region or Line |
|
||||
| `track_thickness` | `int` | `2` | Thickness of Tracking Lines |
|
||||
| `draw_tracks` | `bool` | `False` | Enable drawing Track lines |
|
||||
| `track_color` | `RGB Color` | `(0, 255, 0)` | Color for each track line |
|
||||
| `count_txt_color` | `RGB Color` | `(255, 255, 255)` | Foreground color for Object counts text |
|
||||
| `region_thickness` | `int` | `5` | Thickness for object counter region or line |
|
||||
| `fontsize` | `float` | `0.6` | Font size of counting text |
|
||||
|
||||
### Arguments `model.track`
|
||||
|
||||
| Name | Type | Default | Description |
|
||||
|-----------|---------|----------------|-------------------------------------------------------------|
|
||||
| `source` | `im0` | `None` | source directory for images or videos |
|
||||
| `persist` | `bool` | `False` | persisting tracks between frames |
|
||||
| `tracker` | `str` | `botsort.yaml` | Tracking method 'bytetrack' or 'botsort' |
|
||||
| `conf` | `float` | `0.3` | Confidence Threshold |
|
||||
| `iou` | `float` | `0.5` | IOU Threshold |
|
||||
| `classes` | `list` | `None` | filter results by class, i.e. classes=0, or classes=[0,2,3] |
|
||||
| `verbose` | `bool` | `True` | Display the object tracking results |
|
||||
|
|
@ -318,6 +318,7 @@ nav:
|
|||
- VisionEye Mapping: guides/vision-eye.md
|
||||
- Speed Estimation: guides/speed-estimation.md
|
||||
- Distance Calculation: guides/distance-calculation.md
|
||||
- Queue Management: guides/queue-management.md
|
||||
- YOLOv5:
|
||||
- yolov5/index.md
|
||||
- Quickstart: yolov5/quickstart_tutorial.md
|
||||
|
|
|
|||
187
ultralytics/solutions/queue_management.py
Normal file
187
ultralytics/solutions/queue_management.py
Normal file
|
|
@ -0,0 +1,187 @@
|
|||
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
||||
|
||||
from collections import defaultdict
|
||||
|
||||
import cv2
|
||||
|
||||
from ultralytics.utils.checks import check_imshow, check_requirements
|
||||
from ultralytics.utils.plotting import Annotator, colors
|
||||
|
||||
check_requirements("shapely>=2.0.0")
|
||||
|
||||
from shapely.geometry import Point, Polygon
|
||||
|
||||
|
||||
class QueueManager:
|
||||
"""A class to manage the queue management in real-time video stream based on their tracks."""
|
||||
|
||||
def __init__(self):
|
||||
"""Initializes the queue manager with default values for various tracking and counting parameters."""
|
||||
|
||||
# Mouse events
|
||||
self.is_drawing = False
|
||||
self.selected_point = None
|
||||
|
||||
# Region & Line Information
|
||||
self.reg_pts = [(20, 60), (20, 680), (1120, 680), (1120, 60)]
|
||||
self.counting_region = None
|
||||
self.region_color = (255, 0, 255)
|
||||
self.region_thickness = 5
|
||||
|
||||
# Image and annotation Information
|
||||
self.im0 = None
|
||||
self.tf = None
|
||||
self.view_img = False
|
||||
self.view_queue_counts = True
|
||||
self.fontsize = 0.6
|
||||
|
||||
self.names = None # Classes names
|
||||
self.annotator = None # Annotator
|
||||
self.window_name = "Ultralytics YOLOv8 Queue Manager"
|
||||
|
||||
# Object counting Information
|
||||
self.counts = 0
|
||||
self.count_txt_color = (255, 255, 255)
|
||||
|
||||
# Tracks info
|
||||
self.track_history = defaultdict(list)
|
||||
self.track_thickness = 2
|
||||
self.draw_tracks = False
|
||||
self.track_color = None
|
||||
|
||||
# Check if environment support imshow
|
||||
self.env_check = check_imshow(warn=True)
|
||||
|
||||
def set_args(
|
||||
self,
|
||||
classes_names,
|
||||
reg_pts,
|
||||
line_thickness=2,
|
||||
track_thickness=2,
|
||||
view_img=False,
|
||||
region_color=(255, 0, 255),
|
||||
view_queue_counts=True,
|
||||
draw_tracks=False,
|
||||
count_txt_color=(255, 255, 255),
|
||||
track_color=None,
|
||||
region_thickness=5,
|
||||
fontsize=0.7,
|
||||
):
|
||||
"""
|
||||
Configures the Counter's image, bounding box line thickness, and counting region points.
|
||||
|
||||
Args:
|
||||
line_thickness (int): Line thickness for bounding boxes.
|
||||
view_img (bool): Flag to control whether to display the video stream.
|
||||
view_queue_counts (bool): Flag to control whether to display the counts on video stream.
|
||||
reg_pts (list): Initial list of points defining the counting region.
|
||||
classes_names (dict): Classes names
|
||||
region_color (RGB color): Color of queue region
|
||||
track_thickness (int): Track thickness
|
||||
draw_tracks (Bool): draw tracks
|
||||
count_txt_color (RGB color): count text color value
|
||||
track_color (RGB color): color for tracks
|
||||
region_thickness (int): Object counting Region thickness
|
||||
fontsize (float): Text display font size
|
||||
"""
|
||||
self.tf = line_thickness
|
||||
self.view_img = view_img
|
||||
self.view_queue_counts = view_queue_counts
|
||||
self.track_thickness = track_thickness
|
||||
self.draw_tracks = draw_tracks
|
||||
self.region_color = region_color
|
||||
|
||||
if len(reg_pts) >= 3:
|
||||
print("Queue region initiated...")
|
||||
self.reg_pts = reg_pts
|
||||
self.counting_region = Polygon(self.reg_pts)
|
||||
else:
|
||||
print("Invalid region points provided...")
|
||||
print("Using default region now....")
|
||||
self.counting_region = Polygon(self.reg_pts)
|
||||
|
||||
self.names = classes_names
|
||||
self.track_color = track_color
|
||||
self.count_txt_color = count_txt_color
|
||||
self.region_thickness = region_thickness
|
||||
self.fontsize = fontsize
|
||||
|
||||
def extract_and_process_tracks(self, tracks):
|
||||
"""Extracts and processes tracks for queue management in a video stream."""
|
||||
|
||||
# Annotator Init and queue region drawing
|
||||
self.annotator = Annotator(self.im0, self.tf, self.names)
|
||||
|
||||
if tracks[0].boxes.id is not None:
|
||||
boxes = tracks[0].boxes.xyxy.cpu()
|
||||
clss = tracks[0].boxes.cls.cpu().tolist()
|
||||
track_ids = tracks[0].boxes.id.int().cpu().tolist()
|
||||
|
||||
# Extract tracks
|
||||
for box, track_id, cls in zip(boxes, track_ids, clss):
|
||||
# Draw bounding box
|
||||
self.annotator.box_label(box, label=f"{self.names[cls]}#{track_id}", color=colors(int(track_id), True))
|
||||
|
||||
# Draw Tracks
|
||||
track_line = self.track_history[track_id]
|
||||
track_line.append((float((box[0] + box[2]) / 2), float((box[1] + box[3]) / 2)))
|
||||
if len(track_line) > 30:
|
||||
track_line.pop(0)
|
||||
|
||||
# Draw track trails
|
||||
if self.draw_tracks:
|
||||
self.annotator.draw_centroid_and_tracks(
|
||||
track_line,
|
||||
color=self.track_color if self.track_color else colors(int(track_id), True),
|
||||
track_thickness=self.track_thickness,
|
||||
)
|
||||
|
||||
prev_position = self.track_history[track_id][-2] if len(self.track_history[track_id]) > 1 else None
|
||||
|
||||
if len(self.reg_pts) >= 3:
|
||||
is_inside = self.counting_region.contains(Point(track_line[-1]))
|
||||
if prev_position is not None and is_inside:
|
||||
self.counts += 1
|
||||
|
||||
label = "Queue Counts : " + str(self.counts)
|
||||
|
||||
if label is not None:
|
||||
self.annotator.queue_counts_display(
|
||||
label,
|
||||
points=self.reg_pts,
|
||||
region_color=self.region_color,
|
||||
txt_color=self.count_txt_color,
|
||||
fontsize=self.fontsize,
|
||||
)
|
||||
|
||||
self.counts = 0
|
||||
self.display_frames()
|
||||
|
||||
def display_frames(self):
|
||||
"""Display frame."""
|
||||
if self.env_check:
|
||||
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)
|
||||
# Break Window
|
||||
if cv2.waitKey(1) & 0xFF == ord("q"):
|
||||
return
|
||||
|
||||
def process_queue(self, im0, tracks):
|
||||
"""
|
||||
Main function to start the queue management process.
|
||||
|
||||
Args:
|
||||
im0 (ndarray): Current frame from the video stream.
|
||||
tracks (list): List of tracks obtained from the object tracking process.
|
||||
"""
|
||||
self.im0 = im0 # store image
|
||||
self.extract_and_process_tracks(tracks) # draw region even if no objects
|
||||
|
||||
if self.view_img:
|
||||
self.display_frames()
|
||||
return self.im0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
QueueManager()
|
||||
|
|
@ -363,6 +363,37 @@ class Annotator:
|
|||
cv2.polylines(self.im, [points], isClosed=False, color=color, thickness=track_thickness)
|
||||
cv2.circle(self.im, (int(track[-1][0]), int(track[-1][1])), track_thickness * 2, color, -1)
|
||||
|
||||
def queue_counts_display(self, label, points=None, region_color=(255, 255, 255), txt_color=(0, 0, 0), fontsize=0.7):
|
||||
x_values = [point[0] for point in points]
|
||||
y_values = [point[1] for point in points]
|
||||
center_x = sum(x_values) // len(points)
|
||||
center_y = sum(y_values) // len(points)
|
||||
|
||||
text_size = cv2.getTextSize(label, cv2.FONT_HERSHEY_SIMPLEX, fontScale=fontsize, thickness=self.tf)[0]
|
||||
text_width = text_size[0]
|
||||
text_height = text_size[1]
|
||||
|
||||
rect_width = text_width + 20
|
||||
rect_height = text_height + 20
|
||||
rect_top_left = (center_x - rect_width // 2, center_y - rect_height // 2)
|
||||
rect_bottom_right = (center_x + rect_width // 2, center_y + rect_height // 2)
|
||||
cv2.rectangle(self.im, rect_top_left, rect_bottom_right, region_color, -1)
|
||||
|
||||
text_x = center_x - text_width // 2
|
||||
text_y = center_y + text_height // 2
|
||||
|
||||
# Draw text
|
||||
cv2.putText(
|
||||
self.im,
|
||||
label,
|
||||
(text_x, text_y),
|
||||
cv2.FONT_HERSHEY_SIMPLEX,
|
||||
fontScale=fontsize,
|
||||
color=txt_color,
|
||||
thickness=self.tf,
|
||||
lineType=cv2.LINE_AA,
|
||||
)
|
||||
|
||||
def display_counts(
|
||||
self, counts=None, tf=2, fontScale=0.6, line_color=(0, 0, 0), txt_color=(255, 255, 255), classwise_txtgap=55
|
||||
):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue