Gluon Face Toolkit

GluonFR是一个基于MXNet Gluon API实现的人脸识别工具箱. 此项目灵感来自GluonCV, 并按照其结构组织. 除了帮助研究者和开发者们迅速上手目前最前沿的人脸识别算法, 也希望能够让更多的人了解Gluon这一好用的工具, 使用MXNet-Gluon进行深度学习算法的研究.

提示

如果你不了解MXNet和Gluon, 请参考《动手学深度学习》.

Gluon Face提供目前人脸识别中主流的损失函数, 包括ArcLoss, TripletLoss, RingLoss, CosLoss, L2Softmax, ASoftmax, CenterLoss, ContrastiveLoss等, 在此基础上我们会继续更新. 如果有希望我们实现的损失函数, 欢迎提交 issue.

Gluon Face中的损失函数

下表中最后一列是论文中在LFW上的最优结果, 训练数据/网络结构可能不同, 仅供参考.

Method

Paper

Visualization of MNIST

LFW

Contrastive Loss

ContrastiveLoss

Triplet

1503.03832

99.63±0.09

Center Loss

CenterLoss

img2

99.28

L2-Softmax

1703.09507

99.33

A-Softmax

1704.08063

99.42

CosLoss/AMSoftmax

1801.05599/1801.05599

img3

99.17

Arcloss

1801.07698

img4

99.82

Ring loss

1803.00130

img5

99.52

LGM Loss

1803.02988

img6

99.20±0.03

作者

{ haoxintong Yangxv }

讨论区

我们的讨论区在 Gluon中文论坛

参考文献

  1. MXNet 文档和教程 http://zh.d2l.ai/

  2. NVIDIA DALI 文档NVIDIA DALI documentation

  3. Deepinsight insightface