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

  目录

(一) 人脸数据集对齐

(二) 生成lst文件

---

(一) 人脸数据集对齐

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

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

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

 

(二) 生成lst文件

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

开始制作。首先复制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:训练集和测试集的比例