Model-Swap-Face v2

效果预览

Inference框架

使用方法

环境配置

  参照pSp文档进行环境配置。

下载模型

  模型下载地址参见pretrained_models文件夹中的txt文档，主要包括的模型如下：
  encoder.pt -- 含位姿编码器
  encoder_without_pos.pt -- 去位姿编码器
  projector_EastAsian.pt -- 东亚模特投射器
  projector_WestEuropean.pt -- 西欧模特投射器
  projector_NorthAfrican.pt -- 北非模特投射器
  shape_predictor_68_face_landmarks.dat -- 人脸检测
  79999_iter.pth -- 人脸mask分割

运行代码

  将图片放在input文件夹下，然后编辑scripts/inference.py，运行如下代码：
  python scripts/inference.py
  结果会保存在output文件夹下

了解更多

  我周末的时候喜欢思考一些新的应用场景和模式，欢迎访问我的网站了解更多。