ObjectRecognitionInfoSearch

System for object detection and find information about detected object
Система для распознавания объектов на видео и поиска информации, связанной с найденными объектами

Установка и запуск

При установке данных требований, запуск возможен только на cpu Для запуска на gpu нужно переустановить torch с актуальной версией При запуске система установит модель yolo. По умолчанию yolov5x6u.pt

pip install -r requirements.txt # Установка зависимостей
python main.py # запуск

Для запуска требуется .env файл с настройками Нужно дописать настройки ключа для поиска google, кода поисковой системы и название модели yolo

api_key=''
cx_code=''
model_name='yolov5x6u'

• api_key - ключ от google api для того, чтобы ответ приходил в json формате
• cx_code - ключ поисковой системы google, ее нужно создать для своей системы

Для запуска системы нужно указать путь до видео или 0 для веб-камеры в main.py

video_path = "test_videos/streets_nyc.mp4"  # - path to video
video_path = 0  # - vebcam

Работа системы

При запуске открывается окно с проигрывателем видео, в котором отображаются распознанные объекты с указанием положения, названия объекта и точностью В консоль выводятся данные по распознанным объектам и данные по найденным объектам после поиска

В видео для каждого кадра происходит распознавание объектов и поиск информации по объектам:

• Найденные объекты на видео отображаются с названием класса и точностью
• Найденная информация для объектов выводятся в лог

Пример работы системы: