老照片修复中面临着许多的图像处理问题，比如填孔洞、去划痕、上色、去噪等，也就是说包含了多种图像退化问题。
Bringing-Old-Photo-Back-to-life是由微软研究团队开发的一款修复旧照片的工具，用PyTorch实现，支持划痕检测、面部修饰、全局还原等操作。下面是官方展示的一张修复样张。效果还是很不错的。
左边是待修复的图片,右边是修复完成的照片。
GitHub数据
-
10.8k stars -
264 watching -
1.4k forks
开源地址:https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life
环境要求
该代码在安装了 Nvidia GPU 和 CUDA 的 Ubuntu 上进行了测试。运行代码需要 Python>=3.6。
安装
安装Synchronized-BatchNorm-PyTorch
cd Face_Enhancement/models/networks/
git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch
cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm .
cd ../../../
cd Global/detection_models
git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch
cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm .
cd ../../
下载模型
cd Face_Detection/
wget http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2
bzip2 -d shape_predictor_68_face_landmarks.dat.bz2
cd ../
下载预训练好的模型，把文件放在
Face_Enhancement/checkpoints.zip下./Face_Enhancement，文件Global/checkpoints.zip放在./Global. 然后分别解压。
cd Face_Enhancement/
wget https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life/releases/download/v1.0/face_checkpoints.zip
unzip face_checkpoints.zip
cd ../
cd Global/
wget https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life/releases/download/v1.0/global_checkpoints.zip
unzip global_checkpoints.zip
cd ../
安装依赖项
pip install -r requirements.txt
使用案例
没有划痕的图像
python run.py --input_folder [test_image_folder_path]
--output_folder [output_path]
--GPU 0
划伤的图像
python run.py --input_folder [test_image_folder_path]
--output_folder [output_path]
--GPU 0
--with_scratch
带有划痕的高分辨率图像
python run.py --input_folder [test_image_folder_path]
--output_folder [output_path]
--GPU 0
--with_scratch
--HR
划痕检测
cd Global/
python detection.py --test_path [test_image_folder_path]
--output_dir [output_path]
--input_size [resize_256|full_size|scale_256]
图形用户界面操作
上面都是命令行操作,官方也提供了gui界面操作。
-
-
运行 GUI.py 文件。 -
-
单击浏览并从 test_images/old_w_scratch 文件夹中选择您的图像以删除划痕。 -
-
单击修改照片按钮。 -
-
稍等片刻，在 GUI 窗口中查看结果。 -
-
通过单击退出窗口退出窗口并在输出文件夹中获取结果图像。
更多功能广大网友可以继续挖掘。
原文始发于微信公众号（开源日记）：10.8k stars老照片修复神器
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/49874.html
