d1ffic00lt / ai-tracking

AI animals registration

Описание системы

1. Распознавание
2. Классификация
3. Регистрация

Распознавание

Используется обычная модель yolov8n, которая детектирует животных на картинке, на основе этих данных мы берём количество животных в регистрации

Классификация

Вырезаем Bounding Box, который вадала нам yolo, нормализуем, изменяем размер, отдаём в модель, которая выдаёт нам класс

Регистрация

1. Проходим по отсортированным картинкам

   Отслеживаем животных на них и сохраняем время первого появления

   1.1. Получаем разницу между текущим временем и последним появлением

   1.2. Если время > 30 минут, то записываем регистрацию и перестаем отслеживать объект

   1.3. Обновляем текущее отслеживание

   Время, максимальное количество объектов и т.д.

2. Дополняем регистрации неучтенными отслеживаниями

Описания файлов

• Веса моделей
• Веб интерфейс
• Веб интерфейс на Flask
• Реализация регистрации животного
• Ноутбук с генерацией датасета
• Ноутбук с обучением классификатора (VGG19)