[发明专利]一种可见光手掌识别模型的训练方法和装置

说明书

本申请公开了一种可见光手掌识别模型的训练方法和装置，通过扩增后的可见光手掌数据集对第一卷积神经网络进行训练，得到训练好的第一卷积神经网络；将通过预置对抗生成网络生成的无标签可见光手掌数据集输入到训练好的第一卷积神经网络进行软标签预测，得到软标签可见光手掌数据集；结合扩增后的可见光手掌数据集和软标签可见光手掌数据集对第二卷积神经网络进行训练，得到可见光手掌识别模型，解决了现有的小规模可见光手掌数据集训练得到的手掌识别模型的识别精度不高，且继续采集大规模的可见光手掌图像需耗费大量的人力物力，工作量大、耗费时间长的技术问题。

技术领域

本申请涉及图像识别技术领域，尤其涉及一种可见光手掌识别模型的训练方法和装置。

背景技术

生物特征识别技术应用越来越广泛，例如门禁、支付等领域，生物特征识别技术采用的识别特征有人脸、指纹和手掌等。而现有的可见光手掌识别数据集的规模较小，在小规模可见光手掌数据集训练得到的手掌识别模型，存在识别精度不高的问题；而继续采集大规模的可见光手掌图像需耗费大量的人力物力，工作量大、且耗费时间长。

发明内容

本申请提供了一种可见光手掌识别模型的训练方法和装置，用于解决现有的小规模可见光手掌数据集训练得到的手掌识别模型的识别精度不高，且继续采集大规模的可见光手掌图像需耗费大量的人力物力，工作量大、耗费时间长的技术问题。

有鉴于此，本申请第一方面提供了一种可见光手掌识别模型的训练方法，包括：

对有标签可见光手掌数据集进行数据扩增，将扩增后的所述可见光手掌数据集中的手掌图像和对应的标签输入至第一卷积神经网络进行特征提取；

基于提取的特征向量和所述标签，通过第一损失函数计算网络损失，基于所述网络损失更新所述第一卷积神经网络的参数，直至所述第一卷积神经网络收敛，得到训练好的第一卷积神经网络；

将通过预置对抗生成网络生成的无标签可见光手掌数据集中的手掌图像输入至所述训练好的第一卷积神经网络进行软标签预测，得到软标签可见光手掌数据集，所述预置对抗生成网络通过有标签可见光手掌数据集训练得到；

结合扩增后的所述有标签可见光手掌数据集和所述软标签可见光手掌数据集训练第二卷积神经网络，通过第二损失函数计算得到的网络损失更新所述第二卷积神经网络的参数，直至所述第二卷积神经网络收敛，得到可见光手掌识别模型。

可选地，所述预置对抗生成网络包括鉴别器和生成器，所述预置对抗生成网络的配置过程为：

每训练一次所述鉴别器后，训练一次所述生成器，直至所述对抗生成网络收敛，得到所述预置对抗生成网络。

可选地，所述鉴别器的训练过程，包括：

将所述有标签可见光手掌数据集中的若干张真实的手掌图像和若干张假手掌图像分别输入至所述鉴别器，输出所述真实的手掌图像的分数和所述假图像的分数，所述假手掌图像通过所述生成器生成；

基于所述真实的手掌图像的分数和所述假图像的分数，通过真假图像分类损失函数计算真假图像分类损失；

计算所述真假图像分类损失相对于鉴别器参数的梯度，并基于梯度下降法更新所述鉴别器参数。

可选地，所述鉴别器的训练过程，还包括：

每训练第一预置次数的所述鉴别器后，通过梯度正则化损失函数计算所述真实的手掌图像的梯度正则化损失；

计算所述梯度正则化损失相对于所述鉴别器参数的梯度，并基于梯度下降法更新所述鉴别器参数。

可选地，所述生成器的训练过程，包括：

将获取的噪声数据输入所述生成器，输出所述假手掌图像；