Обучение генеративной нейросети под стиль моей бабушки
Описание проекта
Я попытался обучить нейросеть рисовать в стиле моей бабушки. Она отлично рисует, и в детстве её рисунки заменяли мне сказки. Теперь я хочу дать этому стилю цифровое продолжение и показать, как личная история может стать универсальным языком.
Для обучения и генерации изображений я использовал платформы Kaggle и Hugging Face, а также собственный собранный датасет. Мне было интересно проверить, насколько нейросеть сможет отходить от исходных данных, при этом сохраняя характерный стиль.
Подготовка датасета
Несколько примеров бабушкиных работ, взятых в работу
Переносим датасет в rub папку. Инициализируем функцию для отображения изображений.
Собственный датасет, в который вошло 46 различных работ моей бабушки. Эти рисунки легли в основу стиля
Работа с кодом
Генерируем описание к изображению
Подключение токена Hugging Face
Ставим обучаться на 500 шагов, с сохранением каждые 250 шагов.
После обучения у меня вылезла ошибка, которую, я так и не смог решить…
Долгое время, я пытался разобраться с ошибкой, которая возникала при запуске проекта: система выдавала ImportError, связанный с невозможностью найти элемент xp_capabilities в библиотеке scipy. Проблема оказалась в том, что версии scipy и scikit-learn были несовместимы: новая версия scipy уже не содержала нужной функции, а установленный scikit-learn продолжал к ней обращаться. Я пробовал разные подходы — обновлял и переустанавливал библиотеки, ставил последние версии, менял окружения и даже пытался запускать проект в разных средах, но ошибка повторялась.
