人脸识别之insightface 自定义数据集制作

(1)数据对齐并生成lst文件

cd /data/insightface/src/align/

 python align_lfw.py --input-dir ../../datasets/lfw --output-dir ../../datasets/out2

人脸识别之insightface 自定义数据集制作

(2)创建property配置文件

在datasets/train下创建property,没有后缀

写入下面内容,含义1000,112,112代表ID数量,尺寸,尺寸

 

(3)生成rec&idx文件(依托于lst文件)

运行src/data face2rec2.py

运行成功后会出现两个文件

train.idx和train.rec


(4)创建pair文件

为了做测试我们需要生成验证集用的bin文件,bin文件生成前需要做pair文件,就是一对一对的数据,每一行分别是


图A的目录 空格 图B的目录 空格 标志0/1(代表两张图类别一致否)
在src/data下新建一个代码generate_image_pairs.py用于生成pairs
 

(5)生成验证集bin文件

成功后利用/src/data/下的 lfw2pack.py生成bin文件

但是存在点问题,对lfw2pack.py进行稍微的修改,修改lfw2pack.py中19行,打#的为更改的,改为两个参数,一个是txt读出来的列表,另一个是总数量。