Обучение генеративной модели для рисования медведей
идея проекта
Медведи являются одними из самых узнаваемых и интересных животных, которые еще и сильно связаны с нашей историей. Я много слышала от иностранцев о том, что эти могучие животные являются нашими друзьями и братьями, поэтому решила сделать проект связанный с ними и создать их изображения через обучение нейросети, что позволит лучше понять их разнообразие и красоту.
примеры исходных изображений
Для основы и обучения я брала изображения преимущественно бурых медведей
Я использовала в основном изображения, где медведи находились в лесу или со своими малышами
процесс обучения
Начала я с подключения необходимых модулей
Я учила нейросеть создавать изображения бурых медведей, на основе моего небольшого архива
Чтобы работать с моделями и наборами данных на Hugging Face, я использовала функцию notebook_login (). Благодаря ей я вошла в систему Hugging Face Hub с помощью сгенерированного API-ключа и получила доступ ко всем нужным ресурсам для обучения
Здесь я использовала команду, которая запускает процесс, где я дообучаю модель, чтобы она генерировала картинки в нужном мне стиле. Сначала подгружается базовая модель с нужными ресурсами, а затем ей показывают набор примеров картинок с описаниями, чтобы она научилась создавать новые работы в этом стиле
И следом, я выложила обученную модель на Hugging Face
итоги
Получился очень милый спящий медведь, сделано отлично и пропорции медведя отлично сохранены. Есть вопрос к количеству когтей и пальцев на лапах, но у нейросети часто в целом с этим проблемы
Здесь нейросеть решила добавить людей на фон, а также медвежонка, но все получилось хорошо, особенно мне нравятся яркие красные акценты, которые подчеркнули красивую шерсть медведя
Далее такая же красочная и красивая картинка с сидящим медведем, немного нарушена анатомия животного, но не сильно заметно, возможно нейросеть пыталась усадить его на кресло
Здесь немного беда с лапами, так как появилась дополнительная, а также возникли проблемы с мордой мишки, но если смотреть обобщено, получилось хорошо
Отличное изображение, даже рыба получилась классно, очень похоже на реальную фотографию с красивым пейзажем и самим бурым медведем
На этом изображении получились хорошо два медведя, даже сохранены брызги от воды во время бега, что улучшает эффект съемки
С верхними медвежатами что-то произошло, из-за чего они больше похожи на морских свинок или капибар, но нейросеть старалась :)
вывод
Исходя из моего исследования, могу сказать, что это был очень интересный опыт. Возникли трудности с некоторыми частями генерировании, например с лишними конечностями, но я довольно результатом, так как по большей части получились действительно красивые картинки бурых медведей.
