Deepface 通用模型训练教程

准备工作

一、电脑的虚拟内存要设置在空余空间比较大的盘符中,建议设置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

(0)
打赏 微信扫一扫 微信扫一扫
朋远方的头像朋远方
上一篇 2022年10月10日 上午9:45
下一篇 2022年11月10日 下午5:03

相关推荐

发表回复

登录后才能评论