(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的情况.

 

 

 

 

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐