Расчёт расстояния с помощью Ultralytics YOLO26#

Что такое расчёт расстояния?#

Расчёт расстояния — это процесс измерения промежутка между двумя обнаруженными объектами на изображении или видеокадре. В Ultralytics YOLO26 для расчёта расстояния между выделенными пользователем объектами используется центр ограничивающей рамки.

Иллюстрации#

Расчёт расстояния с помощью Ultralytics YOLO26

Преимущества расчёта расстояния#

• Точность локализации: повышает точность пространственного позиционирования в задачах компьютерного зрения.
• Оценка размера: позволяет оценивать размер объектов для лучшего понимания контекста.
• Понимание сцены: улучшает понимание трёхмерной сцены и помогает принимать более обоснованные решения в таких приложениях, как автономные транспортные средства и системы видеонаблюдения.
• Предотвращение столкновений: позволяет системам выявлять возможные столкновения, отслеживая расстояния между движущимися объектами.
• Пространственный анализ: помогает анализировать взаимосвязи и взаимодействия объектов в наблюдаемой среде.

Расчёт расстояний с помощью YOLO26#

Решение DistanceCalculation отслеживает объекты между кадрами и измеряет евклидово расстояние в пикселях между центрами любых двух ограничивающих рамок, выбранных тобой мышью. Запусти пример ниже, затем щёлкни левой кнопкой мыши по двум объектам, чтобы провести соединительную линию и увидеть расстояние; щёлкни правой кнопкой, чтобы сбросить выбор.

import cv2

from ultralytics import solutions

cap = cv2.VideoCapture("path/to/video.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("distance_output.avi", cv2.VideoWriter_fourcc(*"mp4v"), fps, (w, h))

# Инициализация объекта расчёта расстояния
distancecalculator = solutions.DistanceCalculation(
    model="yolo26n.pt",  # путь к файлу модели YOLO26.
    show=True,  # отобразить результат
)

# Обработка видео
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)  # доступ к результату

    video_writer.write(results.plot_im)  # запись обработанного кадра.

cap.release()
video_writer.release()
cv2.destroyAllWindows()  # закрытие всех открытых окон

Аргументы DistanceCalculation()#

Вот таблица с аргументами DistanceCalculation:

В решении DistanceCalculation также можно использовать различные аргументы track.

Кроме того, доступны следующие аргументы визуализации:

Подробности реализации#

Класс DistanceCalculation отслеживает объекты между видеокадрами и рассчитывает евклидово расстояние между центрами выбранных ограничивающих рамок. Когда ты щёлкаешь по двум объектам, решение:

1. Извлекает центры выбранных ограничивающих рамок
2. Рассчитывает евклидово расстояние между этими центрами в пикселях
3. Отображает расстояние на кадре и соединяет объекты линией

В реализации используется метод mouse_event_for_distance для обработки действий мышью: пользователь может выбирать объекты и при необходимости сбрасывать выбор. Метод process обрабатывает кадры по очереди, отслеживает объекты и рассчитывает расстояния.

Применение#

Расчёт расстояния с помощью YOLO26 находит множество практических применений:

• Аналитика розничной торговли: измерение близости покупателей к товарам и анализ эффективности планировки магазина
• Промышленная безопасность: контроль безопасного расстояния между работниками и оборудованием
• Управление дорожным движением: анализ интервалов между транспортными средствами и выявление опасного сближения
• Спортивная аналитика: расчёт расстояния между игроками, мячом и ключевыми точками поля
• Здравоохранение: обеспечение необходимой дистанции в зонах ожидания и отслеживание перемещения пациентов
• Робототехника: помощь роботам в поддержании безопасного расстояния до препятствий и людей

Часто задаваемые вопросы#