English|简体中文
基于insightface开发,可以轻松替换视频或图片中的人脸。支持windows和linux系统,CPU和GPU推理。onnxruntime推理,无需pytorch。
- 2023/9/16 更新动作迁移算法
- 2023/9/14 更新脸部增强算法(GFPGAN)和超分算法(BSRGAN)
克隆代码,安装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虚拟环境:
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
git clone https://github.com/justld/dofaker.git
cd dofaker
pip install -r requirements.txt
pip install onnxruntime # onnxruntime-gpu
所有的权重来自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使用方式只支持单个人脸替换,同时替换多个人脸请使用命令行的方式:
python web_ui.py
命令行的使用方法支持一次性多个人脸替换:
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 | 新的人脸图片路径,用于替换目标图片或视频 |
禁止将本软件用于违反法律、道德,侵权等场合,本软件仅供学习用途,使用本软件造成的一切后果由使用者承担。