Идея проекта
Исходный датасет, который я выбрала, — картинки крокодильчика Свомпи из популярной 10 лет назад мобильной игры. Целью проекта было обучить модель Stable Diffusion XL, используя метод DreamBooth LoRA.
Исходные изображения для обучения
Для обучения модели я выбрала 25 изображений с крокодильчиком Свомпи. На них так же присутствуют другие персонажи, предметы и локации из вселенной игры. Чем больше деталей, тем проще программе понять и проанализировать необходимый стиль.
Результирующая серия изображений
После обучения была получена новая серия изображений, в которых модель сохраняет ключевые черты персонажа. Это позволяет оценить, насколько эффективно подобное дообучение передает индивидуальные особенности образа.
В итоговой генерации видно, как сохранились основные цвета игры. Позы персонажей и композиция картинок в целом, совпадает с референсными фото.
Папка с кодом и полным набором изображений для обучения
Ноутбук с кодом
