Генерация Концептов Авторитарных Плакатов Для Видеоигр

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

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

Первым этапом стал разгон производственных мощностей и сбор необходимых инструментов для создания алгоритма:

Создание датасета и директории:

Создаем массив с картинками...

Делаем токен и задаем caption

Подготовка к тренировочному процессу (здесь мне и пригодился токен от Huggingface)

Задаем параметры для тренировки, а именно разрешение изображения (512), количество этапов (250) и общее количество тренировочных шагов (500). Обучение начинается и займет примерно 50 минут...

Все почти готово...

Алгоритм готов к употреблению, в строке внизу можно вводить промпт и изображение будет сгенерировано!

Первый блин, конечно, пошел комом: Вот результат экспериментального промпта из лекции преподавателя, созданный для прогрева алгоритма: "photo collage in USSRWARPOSTER style, times square, new york"

В результате работы алгоритма у меня получились изображения, отображающие несколько основнных тематик советского Агитпропа: Военная, Политическая, Индустриальная и Сельскохозяйственная. Сохранение живости и выразительности образов стало для меня основной задачей.

Промпт: "photo collage in USSRWARPOSTER style, agricultural poster"

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