相片修复框架-GFPGAN

一 GFPGAN 介绍

       GFPGAN 是一个由腾讯 ARC 团队开发的用于人脸图像生成和优化的 GAN 模型。在github可以找到开源的代码,它由两个主要模块组成:

  • 退化移除模块 (U-Net):用于从低分辨率、低质量的人脸图像中恢复出高质量的人脸图像。

  • 生成式脸部先验模块 (StyleGAN2):用于生成高分辨率、逼真的人脸图像。

GFPGAN 的工作原理是首先使用退化移除模块来恢复出高质量的人脸图像。然后,使用生成式脸部先验模块来生成高分辨率、逼真的人脸图像。

GFPGAN 具有以下优点:

  • 能够生成高分辨率、逼真的人脸图像。
  • 能够对已有的人脸图像进行增强和优化,例如去除皱纹、瑕疵,增强肤色等。
  • 能够控制生成图像的风格和特征,例如改变年龄、性别、肤色等。

GFPGAN 已在多个人脸图像生成和优化任务上得到了验证,并取得了良好的效果。

GFPGAN 具有广泛的应用前景,可用于以下场景:

  • 人脸合成
  • 人脸美化
  • 人脸识别
  • 人脸动画
  • 虚拟现实/增强现实

 

二 GFPGAN 项目搭建

下拉代码:

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

安装项目依赖组件

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

 下载数据模式https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth

下载后的文件放在位置 experiments/pretrained_models/目录,如下图:

GFPGAN 运行项目

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

运行报错解决:

由于采用默认安装,torchvision没有指定版本这与官方的不一致,需要修改以下源码:

#from torchvision.transforms.functional_tensor import rgb_to_grayscale
# 上面代码引入模块不存在,现变更为如下:
from torchvision.transforms.functional import rgb_to_grayscale

修改后如下图: 

 

首次运行要联网,还会下载数据模型,不然会报错如下图:

 如查不联网可以手工下载保存到相应的位置即可,如下图:

 上图中可以看到运行成功。

GFPGAN 自带照片修复后的效果图如下:

相关推荐

  1. Beego框架相关内容

    2024-02-04 19:56:02       31 阅读
  2. Flask和Go框架相比

    2024-02-04 19:56:02       55 阅读
  3. android pdf框架-3,基于recyclerview修改

    2024-02-04 19:56:02       63 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-02-04 19:56:02       91 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-02-04 19:56:02       97 阅读
  3. 在Django里面运行非项目文件

    2024-02-04 19:56:02       78 阅读
  4. Python语言-面向对象

    2024-02-04 19:56:02       88 阅读

热门阅读

  1. Understanding TCP Congestion Control

    2024-02-04 19:56:02       40 阅读
  2. 数据库||数据库相关知识练习题目与答案

    2024-02-04 19:56:02       44 阅读
  3. KMP算法

    2024-02-04 19:56:02       56 阅读
  4. C语言中大小写字母的转化

    2024-02-04 19:56:02       49 阅读
  5. 微端服务器都需要哪些配置?

    2024-02-04 19:56:02       49 阅读
  6. css鼠标悬浮动效

    2024-02-04 19:56:02       50 阅读
  7. 蓝桥杯练习题-幸运数字

    2024-02-04 19:56:02       50 阅读
  8. 时间序列(Time-Series)Embed.py代码解析

    2024-02-04 19:56:02       31 阅读