Обучение генеративной нейросети под стиль американских ретро открыток
ОПИСАНИЕ ИДЕИ
Я обучила генеративную нейросеть, генерирующую изображения в стиле американских ретро-открыток середины XX века. Нейросеть имитирует узнаваемый стиль открыток, то есть яркие, но выцвевшие цвета, зернистость, более того добавляет характерные дефекты печати.
ИСХОДНЫЕ ИЗОБРАЖЕНИЯ
Нейросеть обучена на массиве ретро-открыток, состоящем из 70 изображений в формате JPEG, взятых с платформы, посвященной данным открыткам и Pinterest.
Исходные изображения собраны на сайте: Американский ретро-арт
ПРОЦЕСС ОБУЧЕНИЯ МОДЕЛИ
Для обучения нейросети я использовала Stable Diffusion. А для получения токена использовала Hugging Face. Код выполнялся в Google Collab.
В среде Kaggle я выбрала ускорение GPU P100 и создала датасет «postcard3». Затем он был перенесен в директорий «post3».
Были задействованы методы Lora, позволяющий провести процесс обучения модели, не прибегая к большим вычислительным затратам, и DreamBooth, который отвечал за концепцию и стиль.
СГЕНЕРИРОВАННЫЕ ИЗОБРАЖЕНИЯ
1. Flowers are in a Starbucks coffee cup and New York is outside the window. 2. A girl is working on a laptop on the veranda of a restaurant against the backdrop of the Eiffel Tower
A hamster is sitting in a sock
The cat is lying on the bed and basking in the sun
The plane lands in Moscow
1. Two pugs are sitting on a mat and playing computer games 2. The castle
Michael Jackson stands on stage
The Coca Cola helicopter is flying over planet earth
A large Christmas tree on Red Square in Moscow and fireworks all around
Palm trees and the beach
1. A girl is sitting with a dog on a bench 2. A girl in modern clothes walks down a fashion runway at a fashion show
1. A mouse with glasses sits under the table and drinks cocktail 2. The parrot is sitting on a chair
ВЫВОД
В завершении работы над обучением нейросети, я могу сделать вывод о том, что получаемые изображения действительно имеют стиль ретро, присущий исходным иллюстрациям. Каждой характерны приглушенные оттенки и общая гармоничная цветовая палитра, поэтому могу сказать, что обучение прошло успешно.
