人脸识别模型的训练方法、装置、电子设备和存储介质制造方法及图纸

本发明专利技术涉及人脸识别技术领域，提供一种人脸识别模型的训练方法、装置、电子设备和存储介质。所述人脸识别模型的训练方法包括：获得人脸识别模型的批处理数据量和训练集的类别数；基于随机数构建0‑1分布，生成列数为所述类别数的参数向量；根据所述参数向量调整人脸识别损失函数的固定缩放比例，获得可调缩放向量，并根据所述可调缩放向量获得行数为所述批处理数据量且列数为所述类别数的缩放矩阵；采用所述缩放矩阵对所述人脸识别模型的输出进行重缩放；以及基于重缩放后的输出，监督训练所述人脸识别模型。本发明专利技术通过对人脸识别损失函数的固定缩放比例进行调整，生成能增大类中心向量之间距离的可调缩放向量，提升人脸识别模型的识别准确率。

【技术实现步骤摘要】
人脸识别模型的训练方法、装置、电子设备和存储介质
本专利技术涉及人脸识别
，具体地说，涉及一种人脸识别模型的训练方法、装置、电子设备和存储介质。
技术介绍
ArcFace(AdditiveAngularMarginLossforDeepFaceRecognition，深度人脸识别的加性角度间隔损失)是人脸识别领域的最新技术，其损失函数为ArcFaceloss(加性角度间隔损失函数)，人脸识别模型的识别准确率主要取决于损失函数的设计。ArcFaceloss在传统人脸识别技术的基础上通过添加角度余量(angularmargin)提高类间可分性并加强类内紧度。但是，没有对特征缩放的半径大小做合理的设置，只取固定缩放比例，使所有的特征向量和类中心向量都变成相同的向量长度，即使所有的特征向量和类中心向量都缩放到半径为该固定缩放比例的超球体上。如果在保持添加的角度余量之后，同时维持特征向量和类中心向量之前的长度不变，那么缩放比例应该等于特征向量和类中心向量的长度之积，特征向量是根据输入数据和卷积层的参数决定的，其长度一直在变化；类中心向量是可本文档来自技高网...

【技术保护点】
1.一种人脸识别模型的训练方法，其特征在于，包括步骤：/n获得人脸识别模型的批处理数据量和训练集的类别数；/n基于随机数构建0-1分布，生成列数为所述类别数的参数向量；/n根据所述参数向量调整人脸识别损失函数的固定缩放比例，获得可调缩放向量，并根据所述可调缩放向量获得行数为所述批处理数据量且列数为所述类别数的缩放矩阵；/n采用所述缩放矩阵对所述人脸识别模型的输出进行重缩放；以及/n基于重缩放后的输出，监督训练所述人脸识别模型。/n

【技术特征摘要】
1.一种人脸识别模型的训练方法，其特征在于，包括步骤：
获得人脸识别模型的批处理数据量和训练集的类别数；
基于随机数构建0-1分布，生成列数为所述类别数的参数向量；
根据所述参数向量调整人脸识别损失函数的固定缩放比例，获得可调缩放向量，并根据所述可调缩放向量获得行数为所述批处理数据量且列数为所述类别数的缩放矩阵；
采用所述缩放矩阵对所述人脸识别模型的输出进行重缩放；以及
基于重缩放后的输出，监督训练所述人脸识别模型。


2.如权利要求1所述的训练方法，其特征在于，所述人脸识别模型基于深度卷积神经网络构建，所述人脸识别损失函数是加性角度间隔损失函数。


3.如权利要求1所述的训练方法，其特征在于，根据所述参数向量调整人脸识别损失函数的固定缩放比例的步骤包括：
altered_s＝S+selected_vector*S*2*(1-cosθj)，
其中，altered_s是所述可调缩放向量，所述可调缩放向量的列数为所述类别数，S是所述固定缩放比例，selected_vector是所述参数向量，θj是相邻两类别的类中心向量的夹角，所述类中心向量根据所述人脸识别模型的输出获得。


4.如权利要求1所述的训练方法，其特征在于，根据所述参数向量调整人脸识别损失函数的固定缩放比例的步骤包括：
altered_s＝S+selected_vector*S*2*(1-cos(θyi+m).mean(dim＝0))，
其中，altered_s是所述可调缩放向量，所述可调缩放向量的列数为所述类别数，S是所述固定缩放比例，selected_vector是所述参数向量，cos(θyi+m)是当前特征向量和目标类中心向量的夹角与一角度间隔值之和的余弦值，所述余弦值是行数为所述批处理数据量且列数为所述类别数的矩阵，所述当前特征向量根据所述人脸识别模型的输出获得。


5.如权利要求3或4所述的训练方法，其特征在于，所述固定缩放比例的取值为：S＝64，所述相邻两类别的类中心向量的夹角的取值为：θj＝71.61÷360×2π弧度。


6.如权利要求1所述的训练方法，其特征在于，基于随机数构建0-1分布，生成列数为所述类...

【专利技术属性】
技术研发人员：沈涛，
申请(专利权)人：携程计算机技术上海有限公司，
类型：发明
国别省市：上海;31