Репозитория решения кейса №2 "РЖД Безопасный маршрут" хакатона "ДВФО Окружной Хакатон ИИ" цифрового прорыва 2023 командой "Пошки"
Предложено решение при помощи двух моделей:
- localModel.py
- remoteModel.py
Для определения опасности используется только видео с фронтальной камеры.
localModel.py использует локальные копиии конфигурации и весов модели YOLOv3 и определяет опасность при пересечении границ детектирования пешехода с выделенной на экране областью. Область неизменная от видео к видео и выбрано как среднее местопложение опасности между ними. Сигнализация о детектировании пешехода реализовано в localModel.py. Программа проходит через все видео, обозначенные в data/input/, сигнализирует красным текстом и звуком о нахождении пешехода на обозначенной области и сохраняет случаи в файл result.csv.
remoteModel.py