欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ArcFace/InsightFace使用自己数据训练/验证过程(1)

程序员文章站 2022-07-12 19:41:06
...

  目录

(一) 人脸数据集对齐

(二) 生成lst文件


(一) 人脸数据集对齐

首先下载insightface源码,点击github链接下载,本文代码需在python2.7环境下运行,文章中所需的脚本文件点击下载

在进行训练之前。首先要准备好自己的数据,数据的存放形式如下图所示,每个文件夹代表一个人,文件夹下放置属于这个人的多张照片。

ArcFace/InsightFace使用自己数据训练/验证过程(1)

然后,需要对所有的照片中的人脸进行对齐处理,并把每张图片缩放到112x112。人脸对齐使用MTCNN,insightface中自带的就有MTCNN和其模型文件,拷贝face_alignment_by_mtcnn.py到deploy文件夹下,修改原始人脸数据文件夹和对齐后要保存的文件夹位置,如下图所示,然后直接运行该程序。

ArcFace/InsightFace使用自己数据训练/验证过程(1)

 

(二) 生成lst文件

lst文件中主要包含图片的路径、标签、序号等信息,为后面制作数据文件做准备,lst文件中的信息如下图所示,第一列时打乱的序号,第二列时图片的地址,第三列是该图片所对应的的标签。

ArcFace/InsightFace使用自己数据训练/验证过程(1)

开始制作。首先复制face2rec2_m.py文件到scr/data文件夹下,然后命令行窗口中输入以下命令:

python -u face2rec2_m.py --list True --prefix F:\datasets\20_faces_datasets\ --root F:\datasets\20_faces_clip\ --train-ratio 0.8 --test-ratio 0.2 --recursive True

参数解释:

--list:是否生成lst文件
--prefix:lst文件保存位置
--root:对齐后人脸数据存放位置
--train-ratio/--test-ratio:训练集和测试集的比例