← 返回列表

人脸识别模型训练方法、装置、电子设备及存储介质

申请号: CN202311832529.5
申请人: 苏州元脑智能科技有限公司
申请日期: 2023/12/28

摘要文本

本发明提供一种人脸识别模型训练方法、装置、电子设备及存储介质,涉及数据处理技术领域,包括:获取训练样本集,对于训练样本集中的任一训练样本,将归一化的训练样本输入到预设神经网络模型,利用神经网络前向传播算法,计算第一损失值;基于第一损失值,计算训练样本的梯度值,并根据训练样本的梯度值中全部数值绝对值的平均值,以及预设对抗强度信息,计算训练样本的离散化梯度值;基于训练样本的离散化梯度值,确定训练样本对应的对抗样本,对抗样本携带有对应训练样本的身份标签;基于各个训练样本和训练样本对应的对抗样本,对预设神经网络模型进行训练,在满足预设训练条件的情况下,得到训练好的人脸识别模型。。更多数据:

专利详细信息

项目 内容
专利名称 人脸识别模型训练方法、装置、电子设备及存储介质
专利类型 发明申请
申请号 CN202311832529.5
申请日 2023/12/28
公告号 CN117496582A
公开日 2024/2/2
IPC主分类号 G06V40/16
权利人 苏州元脑智能科技有限公司
发明人 温东超; 梁玲燕; 赵雅倩; 史宏志; 许光远
地址 江苏省苏州市吴中经济开发区郭巷街道官浦路1号9幢

专利主权项内容

1.一种人脸识别模型训练方法,其特征在于,包括:获取训练样本集,其中,所述训练样本集中包括至少一个训练样本,所述训练样本为携带有身份标签的人脸图像训练样本;对于所述训练样本集中的任一训练样本,将归一化的所述训练样本输入到预设神经网络模型,利用神经网络前向传播算法,计算第一损失值;基于所述第一损失值,计算所述训练样本的梯度值,并根据所述训练样本的梯度值中全部数值绝对值的平均值,以及预设对抗强度信息,计算所述训练样本的离散化梯度值;基于所述训练样本的离散化梯度值,确定所述训练样本对应的对抗样本,所述对抗样本携带有对应训练样本的所述身份标签;基于各个所述训练样本和所述训练样本对应的对抗样本,对预设神经网络模型进行训练,在满足预设训练条件的情况下,得到训练好的人脸识别模型;其中,所述训练好的人脸识别模型用于在输入待识别人脸图像的情况下,输出所述待识别人脸图像对应的身份信息。