Tính toán khoảng cách sử dụng Ultralytics YOLO26#

Tính toán khoảng cách là gì?#

Tính toán khoảng cách là quá trình đo khoảng cách giữa hai đối tượng được phát hiện trong một khung hình ảnh hoặc video. Trong trường hợp của Ultralytics YOLO26, tâm của bounding box được sử dụng để tính toán khoảng cách cho các bounding box được người dùng đánh dấu.

Hình ảnh trực quan#

Tính toán khoảng cách sử dụng Ultralytics YOLO26

Ưu điểm của tính toán khoảng cách#

• Độ chính xác định vị: Nâng cao khả năng định vị không gian chính xác trong các tác vụ computer vision.
• Ước tính kích thước: Cho phép ước tính kích thước đối tượng để hiểu ngữ cảnh tốt hơn.
• Hiểu bối cảnh: Cải thiện khả năng hiểu cảnh 3D để ra quyết định tốt hơn trong các ứng dụng như autonomous vehicles và hệ thống giám sát.
• Tránh va chạm: Cho phép hệ thống phát hiện các va chạm tiềm ẩn bằng cách theo dõi khoảng cách giữa các đối tượng đang di chuyển.
• Phân tích không gian: Tạo điều kiện phân tích các mối quan hệ và tương tác của đối tượng trong môi trường được giám sát.

Tính toán khoảng cách với YOLO26#

Giải pháp DistanceCalculation theo dõi các đối tượng qua các khung hình và đo khoảng cách Euclid, tính bằng pixel, giữa tâm của bất kỳ hai bounding box nào bạn chọn bằng chuột. Chạy ví dụ bên dưới, sau đó nhấp chuột trái vào hai đối tượng để vẽ đường nối và đọc khoảng cách; nhấp chuột phải để xóa lựa chọn của bạn.

import cv2

from ultralytics import solutions

cap = cv2.VideoCapture("path/to/video.mp4")
assert cap.isOpened(), "Error reading video file"

# Video writer
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("distance_output.avi", cv2.VideoWriter_fourcc(*"mp4v"), fps, (w, h))

# Initialize distance calculation object
distancecalculator = solutions.DistanceCalculation(
    model="yolo26n.pt",  # path to the YOLO26 model file.
    show=True,  # display the output
)

# Process video
while cap.isOpened():
    success, im0 = cap.read()

    if not success:
        print("Video frame is empty or processing is complete.")
        break

    results = distancecalculator(im0)

    print(results)  # access the output

    video_writer.write(results.plot_im)  # write the processed frame.

cap.release()
video_writer.release()
cv2.destroyAllWindows()  # destroy all opened windows

Đối số DistanceCalculation()#

Dưới đây là bảng chứa các đối số của DistanceCalculation:

Bạn cũng có thể tận dụng nhiều tham số track khác nhau trong giải pháp DistanceCalculation.

Hơn nữa, các tham số trực quan hóa sau đây cũng có sẵn:

Chi tiết triển khai#

Lớp DistanceCalculation hoạt động bằng cách theo dõi các đối tượng qua các khung hình video và tính toán khoảng cách Euclid giữa tâm của các bounding box được chọn. Khi bạn nhấp vào hai đối tượng, giải pháp sẽ:

1. Trích xuất các trọng tâm (điểm trung tâm) của các bounding box đã chọn
2. Tính toán khoảng cách Euclidean giữa các trọng tâm này theo đơn vị pixel
3. Hiển thị khoảng cách trên khung hình với một đường nối giữa các đối tượng

Việc triển khai sử dụng phương thức mouse_event_for_distance để xử lý các tương tác chuột, cho phép người dùng chọn đối tượng và xóa lựa chọn khi cần. Phương thức process xử lý việc xử lý từng khung hình, theo dõi đối tượng và tính toán khoảng cách.

Ứng dụng#

Tính toán khoảng cách với YOLO26 có nhiều ứng dụng thực tế:

• Phân tích bán lẻ: Đo lường khoảng cách của khách hàng tới sản phẩm và phân tích hiệu quả bố trí cửa hàng
• An toàn công nghiệp: Giám sát khoảng cách an toàn giữa công nhân và máy móc
• Quản lý giao thông: Phân tích khoảng cách giữa các phương tiện và phát hiện hành vi bám đuôi
• Phân tích thể thao: Tính toán khoảng cách giữa các cầu thủ, quả bóng và các vị trí chính trên sân
• Chăm sóc sức khỏe: Đảm bảo khoảng cách phù hợp trong khu vực chờ và giám sát sự di chuyển của bệnh nhân
• Robot: Cho phép robot duy trì khoảng cách phù hợp với các chướng ngại vật và con người

Câu hỏi thường gặp#