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