香港城市大学和微软亚洲研究院的“让老照片重现光彩”(Bringing Old Photos Back to Life)项目源代码请见:
https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life
其源代码比较长,在这里做一些简要的分析(不包含模型部分):
(一)主程序:run.py
主程序一共分为4个步骤,分别是:
步骤1:照片整体修复;
步骤2:面部检测,找出可以增强修复的人脸;
步骤3:对检测出的人脸进行增强修复;
步骤4:将增强修复后的人脸融合到原图,完成全部修复。
(二)步骤1,./Global/detection.py 使用 Unet 进行图像分割,检测出划痕,生成mask文件:
./Global/test.py 调用 Pix2PixHDModel_Mapping() 对照片进行整体修复: