Услуги и решения

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

Отрасль проекта

Машинное зрение, Нейронные сети

О проекте

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

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



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



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


ретушь фото с помощью нейросетии

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


ретушь фото с помощью нейросетии

Вас могут заинтересовать

Умный город, Машинное зрение

PYTHON

Расскажите нам о своей задаче
Мы немедленно возьмём её в работу
Станислав Шешунов
hello@nordclan.com