Подсолнухи. Обучение генеративной нейросети
Идея
Подсолнухи — всеми любимые цветы. Они ассоциируются с теплом и солнцем. Мне всегда они очень нравились, и я решила посвятить им свой проект. В этом проекте я обучила генеративную нейросеть распознавать и создавать изображения подсолнухов
Задача проекта — научить модель точно передавать форму, цвет и текстуру этого растения, чтобы получать новые изображения, похожие на реальные фотографии
Цель проекта
Цель проекта — обучить генеративную нейросеть распознавать и создавать изображения конкретного объекта — подсолнуха. Задача заключалась в том, чтобы получить серию реалистичных изображений, похожих на настоящие фотографии
Ход работы
Для обучения я собрала датасет из множества изображений подсолнухов. Я старалась выбирать разные изображения: с разным фоном, ракурсами, чтобы нейросеть могла лучше понять, как подсолнух выглядит в разных условиях
Обучение проводилось в течение 400 шагов, каждые 50 шагов сохранялись промежуточные результаты
Сгенерированные изображения
Заключение
Модель успешно научилась генерировать изображения подсолнухов.
Она смогла передать форму и цвет подсолнухов достаточно точно, изображения получились достаточно реалистичными и разнообразными
Проект показал, что нейросеть может успешно обучаться на небольшом наборе данных и создавать новые варианты знакомого объекта
Код и датасет
