简单几步部署开源旧照修复项目GFPGAN

今天给大家介绍一款腾讯的开源项目GFPGAN-人脸修复工具。相对部署比较简单,可玩性比较高。一起看看吧!

环境

  • Python >= 3.7
  • PyTorch >= 1.7

部署项目

下载项目

proxychains4 git clone https://github.com/TencentARC/GFPGAN.git 

为了加快下载速度,我这里开启了代理。

安装依赖环境

pip install basicsr
pip install facexlib
pip install -r requirements.txt

安装gfpgan

python setup.py develop

下载预训练模型

GFPGANv1.3.pth

https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth

 具体最新模型链接前往GitHub查看

下载好的文件放在\GFPGAN\experiments\pretrained_models

帮助

Usage: python inference_gfpgan.py -i inputs/whole_imgs -o results -v 1.3 -s 2 [options]...

  -h                   show this help
  -i input             Input image or folder. Default: inputs/whole_imgs
  -o output            Output folder. Default: results
  -v version           GFPGAN model version. Option: 1 | 1.2 | 1.3. Default: 1.3
  -s upscale           The final upsampling scale of the image. Default: 2
  -bg_upsampler        background upsampler. Default: realesrgan
  -bg_tile             Tile size for background sampler, 0 for no tile during testing. Default: 400
  -suffix              Suffix of the restored faces
  -only_center_face    Only restore the center face
  -aligned             Input are aligned faces
  -ext                 Image extension. Options: auto | jpg | png, auto means using the same extension as inputs. Default: auto

示例

python inference_gfpgan.py -i inputs/whole_imgs -o results -v 1.3 -s 2

inputs/whole_imgs 为原始图片位置 生成文件夹为results

当然,如果你觉得太麻烦,可以使用在线项目, 项目地址 

https://app.baseten.co/apps/QPp4nPE/operator_views/RqgOnqV


在线效果

可见对后期的修复还是特别好的。

​​

更多精彩文章 欢迎关注我们


请使用浏览器的分享功能分享到微信等