Расчёт расстояния с использованием Ultralytics YOLO26#

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

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

Визуальные материалы#

Расчёт расстояния с использованием Ultralytics YOLO26

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

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

Расчёт расстояния с YOLO26#

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

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

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

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

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

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

Детали реализации#

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

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

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

Применения#

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

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

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