Генеративная нейросеть для ретуши фото

Перед нами стояла необычная задача научить генеративную нейросеть работать за ретушера фотографий. Для этого мы использовали популярную нейросеть Stable Diffusion и модели LoRA, которые вносят нужные изменения в генерацию изображений.

Чтобы получать фотографии нужного стиля, для обучения нейросети мы использовали исходный датасет заказчика. Обучение проходило в несколько этапов: сначала мы “показали” нейросетиоколо 350 изображений и научили ее стилю ретуширования изображений для памятников. После добавили итерации дообучения на 500 фото и на 750 примерах фото. Так нейросеть поняла, какой стиль необходим для обработки фото и смогла его применять к любому изображению.

Чтобы автоматизировать процесс ретуши фото, мы перенесли функционал в понятного любому пользователю чат бота. Нужно запустить бота и отправить ему исходное изображение, после этого его обработает обученная нейросеть и выдаст 4 варианта готовых фотографий.

После выбора подходящего варианта можно загрузить исходник фотографии и использовать его для заказа памятника.