Обучение генеративной нейросети под собственный стиль
Идея проекта
Идея проекта возникла спонтанно, во время работы над одной из картин. Я рисую с детства, но свой стиль, в котором работаю последние полгода, нашла совсем недавно. Сложно сказать, существует ли для него точное название или есть ли другие художники, работающие в подобной манере.
В основном я пишу женские портреты. Главная задача и сложность — добиться разнообразия, используя различные фоны и золотые элементы потали.
Так, я захотела обучить нейросеть писать картины в своём незамысловатом стиле.
Картины собственного написания.
Исходный data-set
Так как картины пишутся достаточно долго, за эти полгода мной было написано всего две, что, конечно, очень мало для data-сета. Тогда на помощь пришли референсы в схожем стиле. Таким образом, мне удалось собрать небольшой data-сет из 24 картин.
Часть изображений из data-сета
Подготовка и процесс обучения модели
Подготовив окружение для обучения генеративной нейросети, я собрала датасет из отобранных картин и загрузила его в kaggle.
Следующим шагом стал автоматизированный процесс создания промтов для каждой картины.
После настройки окружения, я запустила процесс обучения модели.
Модель обучилась и, наконец, я смогла приступить к написанию промтов и генерации изображений.
Результат
Примеры работ сгенерированных обученной моделью
Результатом я осталась очень довольна. Несмотря на небольшой датасет, модель точно определила, какие объекты на картинах должны оставаться черно-белыми, а где нужно добавить золотые элементы. Этот метод не только позволяет создавать красивые изображения в заданном стиле, но и может служить источником вдохновения для моих будущих работ!
