简介:
GFPGAN是腾讯开源的人脸修复算法,旨在为真实世界面部修复提供实用算法。利用预训练好的面部GAN,GFPGAN实现了丰富多样的生成面部,平衡了修复效果的真实性和保真度,超越了其他方法在真实世界图像修复上的准确性和泛化能力。
算法工具:
- Real-ESRGAN: 一般图像恢复算法
- BasicSR: 图像和视频恢复工具箱
- facexlib: 提供人脸关系函数的集合
- HandyView: 基于PyQt5的图像查看器,便于查看和比较
使用方式:
- 官网在线使用(推荐): 访问官网链接进行在线修复操作。https://replicate.com/tencentarc/gfpgan
- 本地构建: 使用提供的开源代码,在满足一定前提条件后,按照指南进行本地构建。
前提条件:
- Linux 发行版系统
- Python >= 3.7(推荐使用 Anaconda 或 Miniconda)
- PyTorch >= 1.7
- NVIDIA GPU + CUDA
使用步骤:
使用Git克隆仓库代码:
<code>git clone https://github.com/TencentARC/GFPGAN.git cd GFPGAN</code>
安装依赖包:
<code>pip install basicsr pip install facexlib pip install -r requirements.txt python setup.py develop</code>
下载预训练模型:GFPGANv1.3.pthshell
<code>wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth -P experiments/pretrained_models</code>
终端执行:将待修复的图片放置到whole_imgs
目录,执行如下命令:
<code>python inference_gfpgan.py -i inputs/whole_imgs -o results -v 1.3 -s 2</code>
模型版本:
GFPGAN提供了三个版本的模型,用户可以根据实际需求选择:
- V1: 带着色的纸模型
- V1.2: 无着色,不需要CUDA扩展,通过预处理使用更多数据进行训练
- V1.3: 基于V1.2,具有更自然的修复效果,在低质量/高质量的输入上获得更好的结果。
效果演示:
© 版权声明
THE END
暂无评论内容