Обучение генеративной модели по стилю Марианны Веревкиной

В рамках проекта анализируются работы Марианны Веревкиной — яркой и многогранной русской художницы.

Картины Марианны отличаются ее экспрессивной техникой, динамичностью и преимущественно абстрактными силуэтами. Ее работы наполнены эмоциональными образами зачастую вдохновленными бытовыми сюжетами. Для поддержания броской акцентной композиции Марианна использует достаточно жесткую технику и материалы — масло, тушь. Практически все картины передают напряжение и направлены на то, чтобы вызвать некую тревогу и внутренний конфликт у человека.

Именно на эти черты я и планирую делать акцент при обучении модели.

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

Обучение модели происходит в несколько этапов:
1. Загрузка библиотек, подключение к среде выполнения в Google Colab, импорт Google диска с архивом изображений для анализа
2. Работа с датасетом — обработка изображений
3. Подготовка к обучению модели
4. Обучение модели, настройка основных характеристик
5. Генерация итоговой серии

1. Загрузка библиотек

Самый первый этап — загрузка всех библиотек и моделей, с помощью которых будет проходить обучение модели. Для этого я подключилась к GPU среде и установила связь между рабочей средой и своим диском, где хранится папка с архивом изображений.

2. Обработка изображений

На этом этапе происходит анализ исходных изображений. Для начала я подключаю конкретную папку со своего диска, вывожу для проверки несколько изображений оттуда. Далее задаю параметры для названия будущей модели, ее расположения на диске и настраиваю качество изображений.

Модель и архив для удобства названы именем автора исходного стиля.

3-4. Подготовка и обучение

Перед этим этапом я очищаю память для лучшей производительности через torch.cuda.empty_cache () и приступаю к подготовке и обучению своей модели.

После регистрации и подключения токена аккаунта на Hugging Face я прописываю параметры для обучения модели:
— задаю направление куда должна выгружаться моя модель через output_dir
— настраиваю качество изображений через resolution
— задаю настройку промта по которому модель будет четче делать генерации в нужном мне стиле через instance_prompt
— выставляю количество шагов с которым обучается модель через max_train_steps

5. Генерация

Финальный этап — загрузка обученной модели и генерация изображений.

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

Как было упомянуто в начале, Марианна часто обращалась к бытовым сюжетам, стараясь подчеркнуть экспрессивность, драматичные мотивы и так далее.

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