Мы представляем модель для автоматического детектирования различных категорий мусора на ленте мусороперерабатывающего завода, запечатленных на 11-канальном видео. Данное приложение является решением для автоматизации процесса сортировки мусора по категориям для дальнейшего перепроизводства переработанного мусора.
Для работы необходим заранее установленный python 3.10+
Для запуска проекта необходимо установить зависимости. Необходимые для работы проекта библиотеки можно посмотреть в файле requirements.txt и установить их вручную. Также можно воспользоваться командой:
$ pip install -r requirements.txt
Видеопоток с камер передается на обученную нейронную сеть, которая маркирует изображение мусора определенной категорией и передает на web-интерфейс оператора, позволяя как в режиме реально времени видеть расположение определенной категории на ленте, так и в формате csv для дальнейшего анализа
Уникальность нашего решения в том, что использовалась генерация стрима конвейера для лучшего обучения и переразметили мусор, поскольку некоторая разметка была некорректна