(已开源)微软出品:老照片自动修复技术

1-模型修复效果

2-代码配置

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

3-网络权重配置

下载模型权重(文末提供下载方式),将文件 Face_Enhancement/checkpoints.zip 放在 ./Face_Enhancement文件夹下, 将文件 Global/checkpoints.zip 放在 ./Global下. 然后分别进行解压,具体命令如下:

cd Face_Enhancement/
unzip checkpoints.zip
cd ../
cd Global/
unzip checkpoints.zip
cd ../

安装python依赖包:

pip install -r requirements.txt

4-使用方法

处理划痕图片

命令如下:

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

注意使用绝对路径,最后的结果会保存在 ./output_path/final_output/文件夹中.

5-模型效果

该模型还能用于图片全局修复以及人脸图像增强,如下图:

具体可参考链接

6-权重下载

代码作者把权重文件托管在国外网站上,总大小接近2GB,目前已将代码和权重打包放入百度网盘和和备用地址和彩云中国移动网盘,方便大家高速下载。关注并进入公众号后台,回复"666"即可获取下载链接。

 

END

扫码关注获取最新AI资讯与实战案例

实用AI客栈

小编微信号 : langu86

posted @ 2022-11-13 22:35  dlhl  阅读(225)  评论(0)    收藏  举报