Tính toán khoảng cách bằ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 ảnh hoặc khung hình video. Với Ultralytics YOLO26, tâm của bounding box được dùng để tính khoảng cách giữa các bounding box do người dùng chọn.

Hình ảnh minh họa#

Tính toán khoảng cách bằng Ultralytics YOLO26

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

• Độ chính xác định vị Precision: Tăng độ chính xác của việc xác định vị trí trong không gian trong các tác vụ thị giác máy tính.
• Ước tính kích thước: Cho phép ước tính kích thước đối tượng để hiểu rõ hơn ngữ cảnh.
• Hiểu cảnh: Cải thiện khả năng hiểu cảnh 3D để đưa ra quyết định tốt hơn trong các ứng dụng như xe tự hành và hệ thống giám sát.
• Tránh va chạm: Cho phép hệ thống phát hiện nguy cơ va chạm bằng cách theo dõi khoảng cách giữa các đối tượng đang chuyển động.
• Phân tích không gian: Hỗ trợ phân tích mối quan hệ và tương tác giữa các đối tượng trong môi trường được giám sát.

Tính khoảng cách bằng YOLO26#

Giải pháp DistanceCalculation theo dõi các đối tượng qua nhiều khung hình và đo khoảng cách Euclid, tính bằng pixel, giữa tâm của hai bounding box bất kỳ mà 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à xem khoảng cách; nhấp chuột phải để xóa lựa chọn.

import cv2

from ultralytics import solutions

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

# Bộ ghi video
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))

# Khởi tạo đối tượng tính toán khoảng cách
distancecalculator = solutions.DistanceCalculation(
    model="yolo26n.pt",  # đường dẫn đến tệp model YOLO26.
    show=True,  # hiển thị đầu ra
)

# Xử lý 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)  # truy cập đầu ra

    video_writer.write(results.plot_im)  # ghi khung hình đã xử lý.

cap.release()
video_writer.release()
cv2.destroyAllWindows()  # đóng tất cả cửa sổ đang mở

Các đối số của DistanceCalculation()#

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

Bạn cũng có thể sử dụng nhiều đối số track trong giải pháp DistanceCalculation.

Ngoài ra, có thể sử dụng các đối số trực quan hóa sau:

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 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 tâm (điểm trung tâm) của các bounding box được chọn
2. Tính khoảng cách Euclid giữa các tâm này theo đơn vị pixel
3. Hiển thị khoảng cách trên khung hình bằng một đường nối giữa các đối tượng

Triển khai sử dụng phương thức mouse_event_for_distance để xử lý 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ý từng khung hình, theo dõi đối tượng và tính khoảng cách.

Ứng dụng#

Tính khoảng cách bằng YOLO26 có nhiều ứng dụng thực tiễn:

• Phân tích bán lẻ: Đo khoảng cách giữa khách hàng và sản phẩm, đồng thời 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 người lao động 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 khoảng cách giữa các cầu thủ, bóng và những vị trí quan trọng 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à theo dõi di chuyển của bệnh nhân
• Robot học: Giúp robot duy trì khoảng cách phù hợp với chướng ngại vật và con người

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