Deepface 通用模型训练教程

朋远方 • 2022年10月23日下午2:04 • 学习笔记 • 3762 views

准备工作

一、电脑的虚拟内存要设置在空余空间比较大的盘符中，建议设置100GB以上的虚拟内存！

二、电脑的显卡建议使用NVIDIA RTX 3000以上

1.将要提取人脸的目标视频（名人的）复制到workspace文件夹，并且更名为data_src.mp4 (后缀不一定是MP4，但是推荐这个格式。)

2.将下载的faceset.pak文件拷贝到workspace/data_dst/aligned文件夹内

3.将下载的RTT model 224/model文件夹中的所有文件（共7个文件）拷贝到 /workspace/model文件夹下面

4.运行 2) extract images from video data_src.bat 从目标视频中提取每帧图片

其中两次都是直接回车。该步骤运行时间长度取决于你机器的配置。

5.运行 4) data_src faceset extract.bat 从目标视频的图片中提取人脸
将Image size (256-2048?:help): 设置为 640
将 Jpeg quality(1-100?:help): 设置为 100
其他选项默认

6.运行 5.XSeg Generic) data_src whole_face mask – apply.bat 将通用 WF 遮罩（mask）应用于 SRC 数据集
选择你机器上的显卡，直接回车

XSeg可用于排除几乎所有障碍物：手、手指、穿孔、疤痕、面部纹身、口腔、舌头、头发、单个发束、眼镜、眼睛瞳孔、眉毛、整个鼻子、嘴唇等。您可以完全控制模型认为什么是人脸，什么不是。

7.运行 5.XSeg) train.bat
Batch_size(2-16?:help): 设置为 8
其他选项都默认直接回车

8.运行 5.XSeg) data_src trained mask – apply.bat

9.运行 4.2) data_src util faceset pack.bat
Delete original files?(y/n): 输入 y
这一步会将头像文件合并，并且删除源文件只留下合并后的faceset文件
这一步之后也可以删除视频中提出的所有图片，节约磁盘空间

原创文章，作者：朋远方，如若转载，请注明出处：https://caovan.com/deepface-tongyongmoxingxunlianjiaocheng/.html