faceswap-pytorch

基于 PyTorch 的 Deepfakes 人脸交换 仅用于学习和研究

这是与使用 Keras 的 https://github.com/joshua-wu/deepfakes_faceswap 相比的 PyTorch 版本。

源代码可直接从 GitHub 页面下载。

源代码、训练图像及训练好的模型（约 300MB）：

https://pan.baidu.com/s/197RIMB_Po96RZNFzV-wqjA 提取码：z4wa

环境要求：

Python == 3.6
pytorch >= 0.4.0

PyTorch 1.0 也支持。为了获得更好的性能，您需要配备现代 GPU 并支持 CUDA。

运行方法：

简单运行时执行 python train.py

如果您不使用预训练模型，大约需要训练 1000 个 epoch 才能看到效果；而训练到 10000 个 epoch 后，结果将与上方图片所示一致。

注：使用 TensorFlow 后端时，PyTorch 的训练速度略快于 Keras。

Faceswap-Deepfake-Pytorch 快速上手指南

本项目是基于 PyTorch 实现的 Deepfakes 换脸工具，仅供学习与研究使用。相比 Keras 版本，PyTorch 版本在训练速度上略有提升。

环境准备

在开始之前，请确保您的系统满足以下要求：

• 操作系统：Linux / Windows (推荐 Linux)
• Python 版本：3.6
• 深度学习框架：PyTorch >= 0.4.0 (已支持 PyTorch 1.0+)
• 硬件加速：强烈建议配备支持 CUDA 的现代 NVIDIA GPU，以获得最佳训练性能。

依赖安装建议： 推荐使用国内镜像源加速 PyTorch 及相关依赖的安装。例如使用清华源安装 PyTorch：

pip install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

安装步骤

1. 克隆源代码 直接从 GitHub 获取最新源码：

   git clone https://github.com/Oldpan/faceswap-pytorch.git
   cd faceswap-pytorch
   

2. 获取资源文件（可选） 如果您希望直接使用预训练模型或获取示例训练图片（约 300MB），可以从百度网盘下载：

   • 链接: https://pan.baidu.com/s/197RIMB_Po96RZNFzV-wqjA
   • 提取码: z4wa

   下载后请将模型文件放置于项目对应目录中。如果不使用预训练模型，则需要从头开始训练。

基本使用

本项目最核心的功能是模型训练。以下是启动训练的最简单命令：

python train.py

使用说明与预期效果：

• 无预训练模型启动：如果您未加载预训练模型，通常需要训练约 1000 个 epoch 后才能看到初步的换脸效果。
• 高质量效果：若要达到 README 演示图中的效果，通常需要训练至 10000 个 epoch 左右。
• 性能提示：在相同配置下，该 PyTorch 版本的训练速度略快于基于 TensorFlow 后端的 Keras 版本。

注意：本工具仅限用于学术研究和教育目的，请勿用于非法用途。