Система обнаружения оружия для обеспечения безопасности
Цель:
Разработка системы детекции оружия в видеопотоке для оперативного выявления угроз и снижения числа возможных жертв в случае вооруженного нападения.
Создать систему, способную в реальном времени обнаруживать человека с оружием (пистолет или автомат) на видеозаписях с камер наблюдения.
Вводные данные:
Для обучения и тестирования использовались:
- Видеозаписи с камер наблюдения заказчика.
- Дополнительные видеоматериалы из открытых источников.
Выходные данные:
Готовая система анализирует видеопоток покадрово и сообщает:
- Наличие людей с оружием на кадре.
- Количество обнаруженного оружия.
Этапы реализации
- Анализ данных и подбор подходов:
- Исследование существующих технологий компьютерного зрения.
- Выбор модели и методов для решения задачи.
- Разметка и предобработка данных:
- Формирование датасета из 6000 изображений.
- Использование инструментов для подготовки данных (Roboflow).
- Моделирование и эксперименты:
- Обучение нейронной сети на основе YOLO (версии 5 и 8).
- Проведение около 50 экспериментов с фиксацией результатов через ClearML.
- Тестирование:
- Интеграция системы с веб-камерой для проверки работы в реальном времени.
Достижения и метрики
- Выбранная модель: YOLOv8.
- Обучение на 6000 изображениях.
- Результаты:
- Precision: 0.97
- Recall: 0.96
- mAP50: 0.98
- mAP50-95: 0.86
Технологический стек
- Модели и фреймворки: YOLOv5, YOLOv8.
- Библиотеки: OpenCV, Pandas, NumPy.
- Инструменты: Roboflow, ClearML.
Результат
Разработанная система успешно выявляет людей с оружием в видеопотоке, обеспечивая заказчику возможность оперативного реагирования на угрозы. Система была протестирована и показала высокую точность в условиях, близких к реальной эксплуатации.
