【技术实现步骤摘要】
一种可见光手掌识别模型的训练方法和装置
本申请涉及图像识别
,尤其涉及一种可见光手掌识别模型的训练方法和装置。
技术介绍
生物特征识别技术应用越来越广泛,例如门禁、支付等领域,生物特征识别技术采用的识别特征有人脸、指纹和手掌等。而现有的可见光手掌识别数据集的规模较小,在小规模可见光手掌数据集训练得到的手掌识别模型,存在识别精度不高的问题;而继续采集大规模的可见光手掌图像需耗费大量的人力物力,工作量大、且耗费时间长。
技术实现思路
本申请提供了一种可见光手掌识别模型的训练方法和装置,用于解决现有的小规模可见光手掌数据集训练得到的手掌识别模型的识别精度不高,且继续采集大规模的可见光手掌图像需耗费大量的人力物力,工作量大、耗费时间长的技术问题。有鉴于此,本申请第一方面提供了一种可见光手掌识别模型的训练方法,包括:对有标签可见光手掌数据集进行数据扩增,将扩增后的所述可见光手掌数据集中的手掌图像和对应的标签输入至第一卷积神经网络进行特征提取;基于提取的特征向量和所述标签,通过第一损失函数计算网络损失,基于所述网络损失更新所述第一卷积神经网络的参数,直至所述第一卷积神经网络收敛,得到训练好的第一卷积神经网络;将通过预置对抗生成网络生成的无标签可见光手掌数据集中的手掌图像输入至所述训练好的第一卷积神经网络进行软标签预测,得到软标签可见光手掌数据集,所述预置对抗生成网络通过有标签可见光手掌数据集训练得到;结合扩增后的所述有标签可见光手掌数据集和所述软标签可见光手掌数据 ...
【技术保护点】
1.一种可见光手掌识别模型的训练方法,其特征在于,包括:/n对有标签可见光手掌数据集进行数据扩增,将扩增后的所述可见光手掌数据集中的手掌图像和对应的标签输入至第一卷积神经网络进行特征提取;/n基于提取的特征向量和所述标签,通过第一损失函数计算网络损失,基于所述网络损失更新所述第一卷积神经网络的参数,直至所述第一卷积神经网络收敛,得到训练好的第一卷积神经网络;/n将通过预置对抗生成网络生成的无标签可见光手掌数据集中的手掌图像输入至所述训练好的第一卷积神经网络进行软标签预测,得到软标签可见光手掌数据集,所述预置对抗生成网络通过有标签可见光手掌数据集训练得到;/n结合扩增后的所述有标签可见光手掌数据集和所述软标签可见光手掌数据集训练第二卷积神经网络,通过第二损失函数计算得到的网络损失更新所述第二卷积神经网络的参数,直至所述第二卷积神经网络收敛,得到可见光手掌识别模型。/n
【技术特征摘要】
1.一种可见光手掌识别模型的训练方法,其特征在于,包括:
对有标签可见光手掌数据集进行数据扩增,将扩增后的所述可见光手掌数据集中的手掌图像和对应的标签输入至第一卷积神经网络进行特征提取;
基于提取的特征向量和所述标签,通过第一损失函数计算网络损失,基于所述网络损失更新所述第一卷积神经网络的参数,直至所述第一卷积神经网络收敛,得到训练好的第一卷积神经网络;
将通过预置对抗生成网络生成的无标签可见光手掌数据集中的手掌图像输入至所述训练好的第一卷积神经网络进行软标签预测,得到软标签可见光手掌数据集,所述预置对抗生成网络通过有标签可见光手掌数据集训练得到;
结合扩增后的所述有标签可见光手掌数据集和所述软标签可见光手掌数据集训练第二卷积神经网络,通过第二损失函数计算得到的网络损失更新所述第二卷积神经网络的参数,直至所述第二卷积神经网络收敛,得到可见光手掌识别模型。
2.根据权利要求1所述的可见光手掌识别模型的训练方法,其特征在于,所述预置对抗生成网络包括鉴别器和生成器,所述预置对抗生成网络的配置过程为:
每训练一次所述鉴别器后,训练一次所述生成器,直至所述对抗生成网络收敛,得到所述预置对抗生成网络。
3.根据权利要求2所述的可见光手掌识别模型的训练方法,其特征在于,所述鉴别器的训练过程,包括:
将所述有标签可见光手掌数据集中的若干张真实的手掌图像和若干张假手掌图像分别输入至所述鉴别器,输出所述真实的手掌图像的分数和所述假图像的分数,所述假手掌图像通过所述生成器生成;
基于所述真实的手掌图像的分数和所述假图像的分数,通过真假图像分类损失函数计算真假图像分类损失;
计算所述真假图像分类损失相对于鉴别器参数的梯度,并基于梯度下降法更新所述鉴别器参数。
4.根据权利要求3所述的可见光手掌识别模型的训练方法,其特征在于,所述鉴别器的训练过程,还包括:
每训练第一预置次数的所述鉴别器后,通过梯度正则化损失函数计算所述真实的手掌图像的梯度正则化损失;
计算所述梯度正则化损失相对于所述鉴别器参数的梯度,并基于梯度下降法更新所述鉴别器参数。
5.根据权利要求2所述的可见光手掌识别模型的训练方法,其特征在于,所述生成器的训练过程,包括:
将获取的噪声数据输入所述生成器,输出所述假手掌图像;
根据所述假手掌图像的分数,通过非饱和分类损失函数计算非饱和分类损失,所述假手掌图像的分数通过所述鉴别器得到;
计算所述非饱和分类损失相对于生成器参数的梯度,并基于梯度下降法更新所述生成器参数。
6.根据权利要求5所述的可见光手掌识别模型的训练方法,其特征在于,所述生成器的训练过程,还包括:
每训练第二预置次数的所述鉴别器后,通过所述假手掌图...
【专利技术属性】
技术研发人员:刘辉,杨奇,
申请(专利权)人:厦门熵基科技有限公司,
类型:发明
国别省市:福建;35
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。