face2face-demo

face2face-demo

这是一个 pix2pix 演示程序,它从面部特征中学习,并将其转换为面部。此外,还提供了一个支持网络摄像头的应用程序,该应用程序可实时将您的面部转换为经过训练的面部。

开始使用

1. 准备环境

# Clone this repo
git clone [email protected]:datitran/face2face-demo.git

# Create the conda environment from file (Mac OSX)
conda env create -f environment.yml

2. 生成训练数据

python generate_train_data.py --file angela_merkel_speech.mp4 --num 400 --landmark-model shape_predictor_68_face_landmarks.dat

输入:

• file 是您要从中创建数据集的视频文件的名称。
• num 是要创建的训练数据的数量。
• landmark-model 是用于检测特征点的面部特征点模型。提供了预先训练的面部特征点模型 这里。 输出:
• 将创建两个文件夹 original 和 landmarks。

如果您想下载我的数据集，这里还有一个我使用的 视频文件 和生成的 训练数据集（已将 400 张图像分割为训练和验证）。

输入: - model-input 是要导入的模型文件夹。 - model-output 是要导出的模型（压缩）文件夹。 输出: - 它返回一个比原始模型文件大小更小的压缩模型。 2. 其次，我们将压缩的模型冻结为一个文件。