人脸识别之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代表
(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
python face2rec2.py ../../datasets/train3/
运行成功后会出现两个文件
train.idx和train.rec
(4)创建pair文件
为了做测试我们需要生成验证集用的bin文件,bin文件生成前需要做pair文件,就是一对一对的数据,每一行分别是
图A的目录 空格 图B的目录 空格 标志0/1(代表两张图类别一致否)
在src/data下新建一个代码generate_image_pairs.py用于生成pairs
python3 generate_image_pairs.py --data-dir ../../datasets/out3 --outputtxt ../../datasets/train3/pairs.txt --num-samepairs 1000
--num-samepairs :代表想生成的相同的对数;同时会生成不相同 的对数,也是这么多。
(5)生成验证集bin文件
成功后利用/src/data/下的 lfw2pack.py生成bin文件
但是存在点问题,对lfw2pack.py进行稍微的修改,修改lfw2pack.py中19行,打#的为更改的,改为两个参数,一个是txt读出来的列表,另一个是总数量。
(6) 识别与验证
/data/insightface/recognition
CUDA_VISIBLE_DEVICES='0' python -u train.py --network r100 --loss arcface --dataset emore
主要是同目录下config.py的配置。

/data/insightface/recognition/eval
calculate_val_far 修复,防止出现分母为0的情况.
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐


所有评论(0)