Original size 1199x1667

Classic stock. Обучение генеративной нейросети

PROTECT STATUS: not protected
4

Описание идеи

В мире дизайна классические фотографии из стока считаются чем-то нелепым, и не используются дизайнерами в своей работе. Но я думаю, что этот стиль контента уже достаточно сильно закрепился в нашем инфополе, поэтому такое явление как «стоковая фотография» нельзя оставить без внимания. Обычно такие изображения включают в себя следующие характеристики:

• банальная композиция • слишком наигранные эмоции • идеально выглаженные объекты • неестественная обстановка

big
Original size 3500x933

Некоторые изображения из датасета

Поскольку это уже довольно привычный и понятный для всех стиль, я решила, что небольшого датасета будет достаточно для обучения генеративной нейросети.

Описание процесса обучения

Перед началом работы я загрузила все необходимые мне библиотеки. После этого, предоставила доступ к моим изображениям, которые лежали в папке в гугл-диске.

big
Original size 1774x698

Загрузка необходимых библиотек

Для изображений я выбрала разрешение 768 на 768 пикселей, выставила максимальный шаг тренинга на 200 и установила чекпоинт на 250. Таким образом, обучение нейросети прошло достаточно быстро и весь процесс занял 40 минут.

big
Original size 897x794

Параметры изображений

Original size 1735x1019

Фрагмент кода

Original size 784x296

Запуск обучения

После того, как код был реализован, готовая модель была сохранена в папку на моём гугл-диске. Теперь можно было приступать к генерации новых изображений на её основе.

Генерация изображений

Original size 1310x767

Поиск модели в гугл-диск

Original size 1062x1080

Генерация изображений

Original size 1512x553

Процесс генерации изображений

0

Изображения, сгенерированные по моей модели

В результате получились такие генерации. Не сказать, что они идеальные, но это мой первый опыт в обучении нейросети, поэтому думаю, нейросеть уловила основную стилистику стоковых фотографий. Генерация 12 изображений заняла у искусственного интеллекта порядка 5 минут.

Classic stock. Обучение генеративной нейросети
4