Приложение разработано на языке Kotlin. Оно использует модели YOLO11 для классификации и детекции, преобразованные в формат TensorFlow Lite. Изначально приложение запускается в режиме классификации, смена режима производится по нажатию соответствующей кнопки на экране.
Приложение может распознавать следующие виды бытовых отходов:
- Алюминиевая банка
- Пластиковая бутылка
- Полиэтиленовый пакет
- Органические отходы
- Бумага
- Картон
- Стеклянная бутылка
- Консервная банка
- Контейнер для бытовой химии
- Пластиковый стакан
Для обучения моделей используется датасет YOLO Waste Detection by ProjectVerba.
Обучение модели производится на платформе Google Colab.
- Исследовать фотографии в наборе данных
- Исключить классы, содержащие слишком мало фотографий
- Расширить датасет собственными образцами
- Конвертировать модели в формат TensorFlow Lite
- Разработать мобильное приложение для классификации и детекции отходов
Режим классификации:
Режим детекции:
