Обнаружение драк и насилия с YOLOv8

О сервисе

Краткий обзор Данный репозиторий содержит набор данных и модели YOLOv8 (nano и small), обученные для обнаружения драк/насилия и отсутствия насилия/драк как на видео, так и на изображениях. Модели оптимизированы для систем видеонаблюдения и безопасности, где критически важно обнаружение физических столкновений. Набор данных Классы набора данных: Набор данных состоит из двух классов: Насилие/Драка: Случаи, когда присутствует физическое насилие. БезНасилия/БезДраки: Случаи без физических столкновений. Формат данных: Видео и изображения размечены соответствующим образом для каждого класса. Набор данных предназначен для обучения моделей глубокого обучения, таких как YOLOv8 для обнаружения насилия. Модели Модели YOLOv8: Мы в основном обучали модели YOLOv8-nano и YOLOv8-small . Эти модели являются легковесными и эффективными, что делает их подходящими для задач обнаружения в реальном времени в условиях ограниченных ресурсов. Назначение Модели обучены точно определять насильственные события в различных условиях, включая толпы, общественные места и спортивные мероприятия. Ключевые особенности: Обнаружение одного класса: Приведенный код специально разработан для обнаружения одного класса за раз, с акцентом на класс Насилие/Драка . Если цель состоит в обнаружении только Насилие/Драка , модели и код предварительно настроены для этой задачи. События без насилия игнорируются во время обнаружения, что позволяет модели сосредоточиться исключительно на выявлении насильственных действий. Начало работы Предварительные требования: Python 3.8 или выше YOLOv8 (Ultralytics) PyTorch OpenCV Примечания Производительность модели: Модели обучены на разнообразном наборе изображений для обобщения в различных средах. Однако, в зависимости от вашего конкретного случая использования, может потребоваться дополнительная тонкая настройка. Будущие улучшения: Мы планируем расширить набор данных и включить более разнообразные сценарии для повышения точности обнаружения, включая спорт, общественные собрания и многое другое. Ссылки Musawer14/fight_detection_yolov8 Источник: https://qubu.ai/models/obnaruzhenie-drak-i-nasiliya-s-yol-ov8 License: MIT