Концепция
Идея генерации собирательного образа советского агитплаката пришла мне в голову не сразу, но учитывая мою специализацию и то, как часто видеоигры затрагивают политическую тематику я решил создать алгоритм, который, хоть и не является заменой художникам и концепт-артистам, может попробовать передать компоновку и цели плаката для последующей доработки художником проекта.
В качестве основной тематики мною были взяты военные и иные агитационные плакаты первой половины XX века за их выразительность и четкое донесение мысли, которое, надеюсь, смог сохранить и мой алгоритм.
КОД (и он даже работает)
Первым этапом стал разгон производственных мощностей и сбор необходимых инструментов для создания алгоритма:
Создание датасета и директории:
Создаем массив с картинками…
Создание массива с картинками
Делаем токен и задаем caption
Подготовка к тренировочному процессу (здесь мне и пригодился токен от Huggingface)
Задаем параметры для тренировки, а именно разрешение изображения (512), количество этапов (250) и общее количество тренировочных шагов (500). Обучение начинается и займет примерно 50 минут…
Все почти готово…
Алгоритм готов к употреблению, в строке внизу можно вводить промпт и изображение будет сгенерировано!
Результат
Первый блин, конечно, пошел комом: Вот результат экспериментального промпта из лекции преподавателя, созданный для прогрева алгоритма: «photo collage in USSRWARPOSTER style, times square, new york»
«photo collage in USSRWARPOSTER style, a poster»
В результате работы алгоритма у меня получились изображения, отображающие несколько основнных тематик советского Агитпропа: Военная, Политическая, Индустриальная и Сельскохозяйственная. Сохранение живости и выразительности образов стало для меня основной задачей.
Промышленная и Сельскохозяйственная тематика
Промпт: «photo collage in USSRWARPOSTER style, agricultural poster»
дальнейшие промпты будут находиться в подписи к изображению
«photo collage in USSRWARPOSTER style, an agricultural poster»
«photo collage in USSRWARPOSTER style, a field of wheat»
