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

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

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

Hình ảnh minh họa#

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

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

• Độ chính xác định vị: Nâng cao độ chính xác của việc xác định vị trí 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 về 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ư phương tiện tự hành và hệ thống giám sát.
• Tránh va chạm: Cho phép các hệ thống phát hiện nguy cơ va chạm bằng cách giám sát 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 các 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"

# 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

Các argument của DistanceCalculation()#

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

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

Ngoài ra, các đối số trực quan hóa sau đây cũng khả dụng:

Chi tiết triển khai#

Class 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 (các điểm trung tâm) của những 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 cùng với một đường nối giữa các đối tượng

Phần triển khai sử dụng method 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. Method process xử lý việc phân tích 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 tế:

• Phân tích bán lẻ: Đo khoảng cách của khách hàng đến sản phẩm và phân tích mức độ hiệu quả của bố cục 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ủ, quả 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à giám sát chuyển động của bệnh nhân
• Robotics: Cho phép robot duy trì khoảng cách phù hợp với chướng ngại vật và con người

FAQ#