DoFaker: 一个简单易用的换脸工具

基于insightface开发，可以轻松替换视频或图片中的人脸。支持windows和linux系统，CPU和GPU推理。onnxruntime推理，无需pytorch。

更新

2023/9/16 更新动作迁移算法
2023/9/14 更新脸部增强算法(GFPGAN)和超分算法(BSRGAN)

教程

B站视频使用教程
AiStudio在线免费体验

快速开始

克隆代码，安装dofaker

git clone https://github.com/justld/dofaker.git
cd dofaker
conda create -n dofaker
conda activate dofaker
pip install onnxruntime # onnxruntime-gpu
pip install -e .

打开web服务（权重自动下载）:

dofaker

命令行：

bash test.sh

源码安装

手动安装onnxruntime或onnxruntime-gpu.

conda install

创建conda虚拟环境:

git clone https://github.com/justld/dofaker.git
cd dofaker
conda create -n dofaker
conda activate dofaker
pip install -r requirements.txt
pip install onnxruntime # onnxruntime-gpu

pip install

git clone https://github.com/justld/dofaker.git
cd dofaker
pip install -r requirements.txt
pip install onnxruntime # onnxruntime-gpu

二、Download Weight

所有的权重来自release，权重来自底部的链接。

解压下载好的权重文件，目录结构如下所示:

|-dofaker
|-docs
|-weights
----|-models
--------|-buffalo_l
----------|-1k3d68.onnx
----------|-2d106det.onnx
----------|-...
--------|-buffalo_l.zip
--------|-inswapper_128.onnx
--------|-GFPGANv1.3.onnx
--------|-bsrgan_4.onnx
|-run_faceswapper.py
|-web_ui.py

三、使用

您可以以web或命令行的方式进行使用

web ui

web使用方式只支持单个人脸替换，同时替换多个人脸请使用命令行的方式：

python web_ui.py

command

命令行的使用方法支持一次性多个人脸替换：

python run_faceswapper.py --source "image or video path to be swapped" --dst_face_paths "dst_face1_path" "dst_face2_path" ... --src_face_paths "src_face1_path" "src_face2_path" ...

以下的命令会使用src_face1和src_face2替换视频input_video.mp4中的dst_face1和dst_face2 ：

python run_faceswapper.py --source input_video.mp4 --dst_face_paths dst_face1.jpg dst_face2.jpg --src_face_paths src_face1.jpg src_face2.jpg

参数	说明
source	需要替换人脸的图片或视频
dst_face_paths	待替换的图片或视频中的目标人脸路径，如果为None，待替换的图片和视频中的所有人脸都被替换为src_face
src_face_paths	新的人脸图片路径，用于替换目标图片或视频