CallMeMaaaybe/Photo_restoration

Denoising, colorizing and upscaling of old photos

Photo_restoration

Сервис по восстановлению старых фотографий

Stack:

Алгоритм работы:

1. Удаление повреждений

Сначала на поврежденной фотографии определяются зоны с царапинами. Построенная маска накладывается на фото и нейросеть заполняет области пустоты

Так же отдельная нейросеть определяет зоны лиц и проводит повышения разрешения конкретно этих зон

2. Колоризация

Колоризацию проводит отдельная нейросеть обученная на старых цветных фотографиях (так что колоризация любой фотографии получается слегка в винтажном стиле)

3. Повышение разрешения.

Операция повышения разрешения происходит на стороннем сервисе через API запрос. На выхое получаете картинку разрешением х8.

https://deepai.org/machine-learning-model/torch-srgan

4. Обрезка фотографии

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

Пример работ

Веса моделей

/photo_restoration/Face_Enhancement/checkpoints (https://drive.google.com/drive/folders/1MXpcVzNmWY3lxiTcsq6JrUqB-XDd_kDD?usp=sharing)

/photo_restoration/Global/checkpoints (https://drive.google.com/drive/folders/1mzzz0ExKpO11srrxZtcvzmR21DxUnrPE?usp=sharing)