Реализовать простой пайплайн машинного обучения для тестирования существующих моделей и разработки новых
Пайплайн должен содержать следующие стадии:
- Загрузка датасета на локальную машину
- Автоматизированная предобработка данных
- Данные должны быть разделены на тренировочные и валидационные
- Данные хранятся в COCO формате.
- Загрузка и обучение моделей
- Должны поддерживаться существующие модели YOLOv5, YOLOv8
- Модели должны иметь одинаковые настраиваемые интерфейсы ввода/вывода
- Инфографика о скорости и качестве работы модели с сохраненной информацией о используемых гиперпараметрах
- Сохранение моделей в виде .onnx для последующего импортирования в проект.