Пока одни художники боятся ИИ, я сделал модель на своих рисунках и отдаю ее бесплатно
Отлично подходит для бесшовных узоров, абстрактных рисунков и стилизованных под акварель образов. Как использовать и как обучить нейросеть на своих картинках?
Скачать модель здесь:
Как использовать .ckpt модель
Качаете файл .ckpt, кидаете в папку с моделями — а дальше все зависит от форка, который вы используете. В InvokeAI, которым пользуюсь я, модели добавляются обычной загрузкой invoke.py и выполнением команды !import_model в консоли.
Как обучить нейросеть на своих картинках, если нет нвидии 4090
У гугла есть замечательный сервис colab — гугл дает бесплатное железо примерно на час. Этого достаточно, чтобы создать модель на основе 15-30-50 изображений.
Сначала идем сюда и знакомимся с колабом (но ничего не трогаем):
Все, что требуется от пользователя, кроме прокликивания каждого окна с кодом и загрузки картинок, — ключ от hugging face.
Регистрируемся на huggingface.co, а затем создаем ключ в настройках
Ключ вставляем сюда:
Отмечаем сохранение модели в гуглдрайв и папку, если хотите (она создастся автоматически)
Далее указываем, как будем вызывать свой стиль или объект
Для меня это «in style of netsvetaev abstract paintings» и «in style of abstract paintings» — но это не так важно, главное, чтобы вы сами это запомнили.
Затем загружаем картинки:
Нужно использовать 512x512 и лучше подготовить их заранее. Что вырезать — решайте сами, но я рекомендую вырезать так, чтобы сохранялась композиция и значимые объекты целиком. Можно сделать несколько отдельных картинок с увеличенным масштабом.
В зависимости от количества изображений, можно изменить количество шагов обучения. Разумное количество — 100 шагов * количество картинок. Для 8 картинок — а хватит и 8 — достаточно 800 шагов. Слишком много тоже плохо: генерации станут слишком контрастными и на них появятся артефакты.
Далее просто запускайте каждый блок в колабе кнопкой Play в левом верхнем углу и здите завершения каждого кода. Когда доберетесь до участка, показанного выше, увидите процесс обучения. Он займет примерно 10 минут + еще 15-30 минут, в зависимости от количества картинок и шагов.
После обучения можно проверить модель на тестовой генерации и сконвертировать в формат .ckpt:
Теперь файл у вас на диске — скачивайте и добавляйте в любимый форк.
Дополнительно можно удалить лишние файлы и завершить процесс
О себе: Меня зовут Артур Нецветаев, я менеджер продуктов, предприниматель и дизайнер интерфейсов. Я помогаю с разработкой интерфейса InvokeAI и пользуюсь им сам с момента создания.