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	新的人臉圖片路徑，用于替換目標圖片或視頻

聲明

禁止將本軟件用于違反法律、道德，侵權等場合，本軟件僅供學習用途，使用本軟件造成的一切后果由使用者承擔。