人脸识别模型训练

github地址：  face-recognition-supervised-by-center-loss

这里介绍我训练的人脸识别模型，放到了github上，各位看官路过的可以点个star

简介

这个仓库我提供了训练人脸深度神经网络的代码，框架使用的是pytorch。损失函数用的是center_loss。同时也提供了triplet_loss的实现。但是根据我的训练经验，用triplet_loss训练没有用center_loss准确度高。所以推荐center_loss。网络我使用的是标准的vgg16。你也可以根据自己的需求换成resnet 或者其他。我相信会获得更高的准确度（大概高两三个百分点的样子）。因为工作需求我选择了vgg16。训练数据我选择的是vggface2(8631个id 及300万照片)。MSLM 有更多的数据，用MSLM 训练我相信结果h会再好一点（大概一两个百分点）。测试数据集我选择的是LFW

训练结果 

vgg16+vggface2:   测试准确度:  0.967     auc:  0.99