腾讯AI模型GFPGAN火上GitHub热榜第一

来自腾讯的研究人员开发了GFP-GAN照片恢复工具。而且必须再次强调的是,这里的AI是对缺失或损坏的照片细节进行一些猜测,它着手改善。这样,在改进照片时,研究人员警告称,在原件和成品之间实际上可能会有轻微的身份变化。

Gitee:https://gitee.com/nywsr/GFPGAN

# 克隆项目
git clone https://github.com/TencentARC/GFPGAN.git
  
# 进入项目
cd GFPGAN

# 安装依赖
pip install basicsr
pip install facexlib
pip install -r requirements.txt
pip install realesrgan

# 安装程序
python setup.py develop

GFPGAN模型需要通过数据集训练得到,由于训练需要使用的数据量和算力非常大,作者团队提供了许多预处理好的模型给普通用户下载,这样我们就能绕过训练这个步骤直接使用模型,下载地址如下:

https://github.com/TencentARC/GFPGAN/releases/download/v0.2.0/GFPGANCleanv1-NoCE-C2.pth

将想要使用的预训练模型放入 experiments/pretrained_models 文件夹下就可以开始使用了。

使用方法非常简单,进入项目目录后输入以下命令:

python inference_gfpgan.py --model_path experiments/pretrained_models/GFPGANv1.pth --test_path inputs/cropped_faces --save_root results