Обучение генеративной нейросети стилю Ханоха Пивена
Ханох Пивен — известный художник, создающий портреты известных людей с использованием подручных предметов. Его работы просты и непредсказуемы, однако герои остаются узнаваемыми. Часто предметы, которые использует художник, никак не связаны с изображенным человеком, что только добавляет абсурдности портрету. Таким образом, Пивен возводит необычные черты людей до абсолюта.
Генерирование изображений в стилистике определенного художника помогает при создании новых идей для портретов.
Цель проекта — обучить модель Stable Diffusion и сгенерировать изображения в стиле Ханоха Пивена
Также интересно понаблюдать за тем, как нейросеть сможет работать с 2д и 3д объектами на одном холсте.
Датасет
Сгенерированные изображения
В итоговой серии изображений представлены портреты людей, сгенерированные ИИ. В качестве промта были использованы как лица известных людей (актеров, артистов, режиссеров), так и общие характеристики на основе профессий, внешних атрибутов и пр.
Тейлор Свифт
1. Полицейский 2. Учитель
Рианна
1. Доктор 2. Программист
Леди Гага
1. Журналистка 2. Художник
Ковбой
1. Мерлин Монро 2. Леонардо Ди Каприо
1. Альберт Эйнштейн 2. Анджелина Джоли
1. Квентин Тарантино 2. Стив Джобс
1. Альфред Хичкок 2. Чарли Чаплин
1. Эминем 2. Канье Уэст
В сгенерированных портретах удалось передать такие черты стиля Ханоха Пивена, как изображение формы головы и волос широкими и небрежными мазками цвета и использование 3д изображений различных предметов для обозначения черт лица
Работа с кодом
Процесс обучения модели происходил в среде Kaggle
Сначала было необходимо переключиться на GPU, чтобы задачи могли выполняться быстрее, а также установить библиотеки с GitHub
Затем была выполнена работа с датасетом. Для обучения модели были выбраны и подготовлены 30 изображений
Были созданы файлы с подписями к изображениям
На данном этапе происходило само обучение модели
Генерация изображений
Блокнот с кодом
Диск с датасетом
