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

接上一篇ArcFace/InsightFace使用自己数据训练/验证过程(1)
文章中所需的脚本文件点击下载。

(三) 生成rec文件

制作proprety

创建文件proprety，注意没有后缀，以文本形式打开，写上以下内容，如下图所示，第一个数字代表类别总数，第二个和第三个数字代表人脸的尺寸。

生成rec文件

进入到face2rec2_m.py所在的目录，然后命令行窗口中输入以下命令：

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

参数解释：

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

然后会在指定目录生成idx和rec文件，idx文件是辅助生成rec的࿰