Найти тему

Генерация изображений Stable Diffusion + DreamBooth

Дипломный проект в Университете Искусственного Интеллекта

Как можно сгенерировать принт для ткани по заданному стилю с помощью возможностей искусственного интеллекта?

Для решения задачи использовалась точная настройка предобученной модели Stable Diffusion/DreamBooth Text2Image.

Для тонкой настройки нейросети был использован готовый ноутбук с репозитория github: fast-DreamBooth colab From https://github.com/TheLastBen/fast-stable-diffusion и 30 изображений размера 512х512 в стиле скандинавского фолка, взятых из открытых источников.

На вход подаем запрос для генерации на английском (часть запроса сформирована неизменной) и отмечаем стиль паттерна: он может быть обычным и бесшовным.

И, в результате, получаем требуемое изображение в стиле скандинавского фолка:

Сравнение сгенерированных изображений до настройки, и после.
Сравнение сгенерированных изображений до настройки, и после.

Также можно проверить принты сгенерированные для бесшовного паттерна:

-3

код можно посмотреть тут: https://github.com/alesoid/NU_Diplom