Мы представляем модель для детекции и трекинга людей; детекции оружия по видео. В основе нашей модели был использоан ансамбль из трх моделей yolov8l обученных на размеченном датасете из ~44к фотографий: https://www.linksprite.com/gun-detection-datasets
Для работы необходим заранее установленный python 3.10+
Для запуска проекта необходимо установить зависимости. Необходимые для работы проекта библиотеки можно посмотреть в файле requirements.txt и установить их вручную. Также можно воспользоваться командой:
$ pip install -r requirements.txt
Видеопоток с камер передается на обученную нейронную сеть, которая выделяет всех людей и оружие на фото. Также каждом человеку присваивается ID