Сбылась мечта блоггера: научила нейроку рисовать себя



Натренила Dreambooth на своих фотках в Colab'е (занимает минут 20).



〰️ Что это?



Это скрипт с дообучением Stable Diffusion модели. Можно обучить на изображениях любой вещи или человека. После этого сеть сможет рисовать похожие объекты в новом контексте ("Mashkka в стиле киперпанк, например").



〰️ Хотите также?



Тогда ловите рецепт:



1️⃣ Взять свои фото (20-30 штук). Подготовить их, — объект должен быть по центру, примерно как фото на паспорт, и привести их к размеру 512 на 512. Это можно легко и быстро сделать здесь.



2️⃣ Положить фото к себе на Google Drive в любую папку.



3️⃣ Открыть Colab. Там нужно последовательно запускать ячейки. В одной из них понадобится ввести токен с 🤗huggingface. Для этого нужно там зарегаться и взять токен. В графе Instance Dir нужно указать путь к фоточкам на Google Drive.



4️⃣ И вуаля, примерно через 20 минут модель сможет генерировать такую красоту!



А примеры хороших затравок для рисования можно найти тут!