项目简介
本项目是一个利用深度学习技术实现的人脸交换工具。用户可以通过简单的命令行操作或图形用户界面,对图片和视频中的人脸进行提取、训练和转换,最终生成具有不同人脸的图片或视频。
项目的主要特性和功能
- 人脸提取:能从图片或视频中提取人脸并进行对齐处理。
- 模型训练:利用提取的人脸数据训练深度学习模型,实现精确的人脸交换。
- 人脸转换:将训练好的模型应用于源图片或视频,生成新面孔的图片或视频。
- 图形用户界面:提供直观的GUI,方便用户操作。
- 视频处理工具:支持视频转换和处理,借助ffmpeg进行视频与图像的相互转换。
安装使用步骤
环境准备
- 确保系统安装Python 3.x。
- 运行
pip install -r requirements.txt
安装所需依赖包。 - 若有现代GPU,安装CUDA以获取最佳性能。
下载源码
从项目仓库下载源码文件。
运行项目
- 提取人脸:运行
python faceswap.py extract
,从src
文件夹提取人脸到extract
文件夹。 - 训练模型:运行
python faceswap.py train
,用两个文件夹中的照片训练模型,模型保存于models
文件夹。 - 转换人脸:运行
python faceswap.py convert
,将新面孔应用到original
文件夹中的照片,结果保存于modified
文件夹。 - 使用GUI:运行
python faceswap.py gui
,通过图形界面操作。
视频处理
使用 python tools.py effmpeg -h
或直接使用 ffmpeg
工具进行视频和图像的转换。
注意事项
- 所有脚本支持
-h
/--help
选项,可查看详细参数说明。 - 重用现有模型比从头开始训练快很多。
- 若训练数据不足,可从长相相似的人开始,再转换数据。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】