京东人脸识别FaceX Zoo

FaceX-Zoo 是京东最近开源的人脸识别pytorch版工具箱, 抽空读了一下技术报告,做个记录.

工程总体结构

总体可分为训练模块、评估模块、附加模块和人脸SDK,目前支持以及将要支持的模块如上图所示,包含人脸识别常用的人脸检测、人脸校准、人脸识别,人脸识别模块开放训练测试。

预处理

预置了常用的图像变换操作,可拓展

训练:resize、normalization、random crop, flip, roration etc

测试:resizin、 norm

训练模块

传统的训练模式与semi-siamese 训练模式

backbone

集成了主流的特征提取backbone,包括 mobilefacenet,resnet,se-resnet,hrnet 等

监督头

实现了一系列主流的人脸识别head,包括 AM-Softmax, ArcFace, AdaCos, AdaM-softmax, circle-loss等

评估准则

支持 LFW,megaface 等常见数据集

复现结果

表一和表二分别对比了各个backbone和监督头在各个数据集上的表现.

shallow face learning

  • 问题定义:对于真实场景而言,人脸识别训练数据集缺少,同一个ID下的训练数据只有几张,这种训练任务叫做 shallow face learning, 不能按照传统方法进行训练,semi-siamese training(SST) 用来解决该问题. 具体训练细节参照代码

masked face recognition

该仓库同时提供针对 masked 的人脸识别.

ref