/Photo_restoration

Denoising, colorizing and upscaling of old photos

Primary LanguageJupyter Notebook

Photo_restoration

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

My Remote Image

Stack:

Python PyCharm Jupyter Notebook TensorFlow OpenCV

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

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

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

My Remote Image

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

My Remote Image

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

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

My Remote Image

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

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

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

My Remote Image

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

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

My Remote Image

Пример работ

My Remote Image

Веса моделей

/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)