Как определить, спелые ли помидоры
Нейронная сеть, определяющая, спелые помидоры или нет. Включает в себя детекцию и классификацию.
Клиент
Владислав З.
Услуга
Детектор спелых помидор
Дата
2021-05-05
Проблема
Написать нейронную сеть, обнаруживающию помидоры на изображении и классифицирующую их на спелые и не спелые.
Решение
Выбрана архитектуру нейронной сети, собран, аугментирован и аннотирован датасет. Обучена нейронная сеть и проанализированы метрики. Подготовлен метод для вывода модели.
Датасет был собран, используя изображения из интернета. Всего было собрано 500 изображений, на которых были помидоры. Чтобы увеличить количество изображений, была применена аугментация. Так, я увеличил картинки в 1.5 раз.
Для аннотирования датасета использовался сервис roboflow:
Для вывода был написан метод на языке Python и подготовлен GUI.
Ниже Вы можете увидеть примеры работы программы:
