Обучение модели Stable Diffusion на лице Анджелины Джоли
Идея
Для выполнения поставленной задачи был создан датасет, в который вошли 20 разнообразных фотографий Анжелины Джоли. Благодаря такому датасету, модель Stable Diffusion сможет воспроизвести в генерациях её внешность.
Датасет
Обучение модели
Обучение происходилj в среде Google Colab — сервис для программирования на Python.
Серия полученных генераций
Полученные генерации: https://disk.360.yandex.ru/d/qgJPgRmsgqOlkQ
Мы взяли датасет LFW (Labeled Faces in the Wild) через scikit-learn
Исходный код
Код в формате ipybn: https://disk.360.yandex.ru/d/y52zRpOA8hEw6w
Использованные нейросети https://huggingface.co/docs/transformers/en/model_doc/blip — BLIP для генерации текстовых описаний к изображениям https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0 — Stable Diffusion XL base 1.0 для генерации изображений
