人脸识别之insightface 自定义数据集制作
(1)数据对齐并生成lst文件
cd /data/insightface/src/align/
python align_lfw.py --input-dir ../../datasets/lfw --output-dir ../../datasets/out2
(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读出来的列表,另一个是总数量。