一种掌静脉检测模型的训练方法、掌静脉检测模型及存储介质技术

技术编号:35568624 阅读:15 留言:0更新日期:2022-11-12 15:52
本发明专利技术提供了一种掌静脉检测模型的训练方法、掌静脉检测模型及存储介质,所述方法包括:采用resNet网络和FPN网络组合形成的神经网络进行训练得到检测模型,其中,所述FPN网络包括自上至下进行逆卷积变换的第一特征金字塔和自下至上进卷积变换的第二特征金字塔,所述第一特征金字塔和所述第二特征金字塔与所述resNet网络具有相同数量且尺寸的特征层,除了最底层外,所述resNet网络的每一层分别连接至所述第一特征金字塔的相应层,且所述resNet网络的最底层分别连接至所述第一特征金字塔的最顶层和所述第二特征金字塔的最顶层。采用本发明专利技术的技术方案,训练得到的掌静脉检测模型检测精度高。检测精度高。检测精度高。

【技术实现步骤摘要】
一种掌静脉检测模型的训练方法、掌静脉检测模型及存储介质


[0001]本专利技术涉及图像识别
,具体为一种掌静脉检测模型的训练方法、掌静脉检测模型及存储介质。

技术介绍

[0002]掌静脉检测在掌静脉识别系统中具有非常重要的作用,即只有检测准确而且迅速才能保证后续的识别准确性以及在终端的可应用。现有掌静脉检测技术主要分为两类,第一类是基于传统的手动设计的特征,同时结合定位,得到掌静脉的检测结果;第二类方法则是基于深度学习自动提取特征,将分类l oss和定位l oss进行结合,训练得到检测模型。
[0003]现有的掌静脉检测技术,无论是早期的传统方法还是最新的深度学习提取掌静脉特征,由于训练数据采集的原因,对于距离很远或距离很近的手掌的掌静脉检测效果都不好;另外,为了同时保证模型的精度和速度,只能在两者之前取平衡,然而最终的结果导致既不能保证模型的精度,也不能保证模型的速度。

技术实现思路

[0004]本专利技术的目的在于提供一种掌静脉检测模型的训练方法、掌静脉检测模型及存储介质。
[0005]本专利技术实施例中,提供了一种掌静脉检测模型的训练方法,其包括:采用resNet网络和FPN网络组合形成的神经网络进行训练得到检测模型,其中,所述FPN网络包括自上至下进行逆卷积变换的第一特征金字塔和自下至上进卷积变换的第二特征金字塔,所述第一特征金字塔和所述第二特征金字塔与所述resNet网络具有相同数量且尺寸的特征层,除了最底层外,所述resNet网络的每一层分别连接至所述第一特征金字塔的相应层,且所述resNet网络的最底层分别连接至所述第一特征金字塔的最顶层和所述第二特征金字塔的最顶层。
[0006]本专利技术实施例中,在训练之前,对训练图片进行裁剪和缩放。
[0007]本专利技术实施例中,将所述resNet网络的卷积堆结构设置为[3,3,9,3]。
[0008]本专利技术实施例中,在训练时,将所述检测模型的每个特征层中的卷积核的数量减少为原来的一半。
[0009]本专利技术实施例中,在训练样本集中,将五个手指两两之间的最低点作为训练的检测关键点,将其回归的loss与掌静脉检测的loss进行合并,得到最终的loss。
[0010]本专利技术实施例中,最终的loss值L的计算方式如下:
[0011][0012]其中,λ1和λ2分别代表不同损失的权重系数,为小于1的常数;L
cls
为掌静脉分类损失函数,p
i
和分别为预测和真实掌静脉分类;L
box
为掌静脉定位回归损失函数,t
i
和分
别为预测和真实掌静脉方框坐标;L
pts
为掌静脉关键点回归损失函数,l
i
和分别为预测和真实掌静脉关键点坐标。
[0013]本专利技术实施例中,掌静脉分类损失函数
[0014]其中,N是输入的样本数或者实例的数量,i是某一个样本或者实例;M表示样本可能的分类数量,j表示某一个分类;表示某个样本i属于分类j的标签,p
ij
表示样本i预测为分类j的概率;
[0015]掌静脉定位回归损失函数
[0016]其中表示某个样本i的检测框第j个真实坐标位置,x
j
、y
j
表示某个样本i的检测框第j个预测坐标位置;
[0017]掌静脉关键点回归损失函数
[0018]其中表示某个样本i的第j个关键点真实坐标位置,x
j
、y
j
表示某个样本i的第j个关键点预测坐标位置。
[0019]本专利技术实施例中,还提供了一种掌静脉检测模型,其通过上述的掌静脉检测模型的训练方法训练得到。
[0020]本专利技术实施例中,还提供了一种存储介质,其存储有计算机程序代码,所述计算机程序代码被执行时,实现上述的掌静脉检测模型的训练方法。
[0021]与现有技术相比较,采用本专利技术的掌静脉检测模型的训练方法,将FPN网络引入resNet网络形成用于训练掌静脉检测模型的神经网络,并在FPN网络中加入更多底层特征的直接连接,从而减小最底层的特征的信息损失,提高了掌静脉检测模型的精度;进一步地,通过对训练图片进行剪裁或缩放,解决了距离远近导致识别不准的问题;进一步地,对每个卷积层中的卷积核的数量进行0.5倍变换,提高了训练的速度;进一步地,把手指间最低点作为关键点加入训练,也提高了检测模型的精度。
附图说明
[0022]图1是本专利技术实施例的掌静脉检测模型的结构示意图。
[0023]图2是本专利技术实施例将五个手指两两之间的最低点作为训练的检测关键点的示意图。
具体实施方式
[0024]为了使本专利技术的目的、技术方案及优点更加清楚明白,以下结合附图及实施例,对本专利技术进行进一步详细说明。应当理解,此处所描述的具体实施例仅仅用以解释本专利技术,并不用于限定本专利技术。
[0025]以下结合具体实施例对本专利技术的实现进行详细描述。
[0026]如图1所示,本专利技术实施例中,提供了一种掌静脉检测模型的训练方法,采用resNet网络1和FPN网络2组合形成的神经网络进行训练得到检测模型。其中,所述FPN网络2包括自上至下进行逆卷积变换的第一特征金字塔21和自下至上进卷积变换的第二特征金字塔22,所述第一特征金字塔21和所述第二特征金字塔22具有相同的成熟,且所述第一特
征金字塔21的每一层分别连接至所述第二特征金字塔的相应层。
[0027]所述resNet网络1有自下而上尺寸逐步减小的多个特征层组成。所述第一特征金字塔21和所述第二特征金字塔22与所述resNet网络1具有相同数量且尺寸的特征层。除了最底层外,所述resNet网络1的每一层分别连接至所述第一特征金字塔21的相应层,且所述resNet网络1的最底层分别连接至所述第一特征金字塔21的最顶层P5和所述第二特征金字塔22的最顶层N5。
[0028]需要说明的是,在图1中,所述resNet网络1、所述FPN网络2的第一特征金字塔21和第二特征金字塔22均为4层,也可以根据具体需要设计更多的成熟,本专利技术对此不进行限制。
[0029]常规的resNet网络的卷积堆结构为[3,4,6,3],本专利技术实施例中,将所述resNet网络的卷积堆结构设置为[3,3,9,3],可以进一步提升计算精度。
[0030]本专利技术实施例中,采用所述检测模型进行训练之前,对训练图片进行裁剪和缩放处理,从而在训练集中更多的正样本,提高所述检测模型对远距离或者近距离拍摄的手掌的掌静脉检测效果。
[0031]为提高训练的速度,本专利技术实施例中,在训练时,对所述检测模型中每个特征层中的卷积核的数量减少一半。例如,之前模型某一层有128个卷积核,现在调整为64,模型的每一层都要调整,比如假设原来模型一共三层,分别是256,128,128,则现在调整为128,64,64。
[0032]如图2所示,本专利技术实施例中,在训练本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种掌静脉检测模型的训练方法,其特征在于,包括:采用resNet网络和FPN网络组合形成的神经网络进行训练得到检测模型,其中,所述FPN网络包括自上至下进行逆卷积变换的第一特征金字塔和自下至上进卷积变换的第二特征金字塔,所述第一特征金字塔和所述第二特征金字塔具有与所述resNet网络相同数量且尺寸的特征层,除了最底层外,所述resNet网络的每一层分别连接至所述第一特征金字塔的相应层,且所述resNet网络的最底层分别连接至所述第一特征金字塔的最顶层和所述第二特征金字塔的最顶层。2.如权利要求1所述的掌静脉检测模型的训练方法,其特征在于,在训练之前,对训练图片进行裁剪和缩放。3.如权利要求1所述的掌静脉检测模型的训练方法,其特征在于,将所述resNet网络的卷积堆结构设置为[3,3,9,3]。4.如权利要求1所述的掌静脉检测模型的训练方法,其特征在于,在训练时,将所述检测模型的每个特征层中的卷积核的数量减少为原来的一半。5.如权利要求1所述的掌静脉检测模型的训练方法,其特征在于,在训练样本集中,将五个手指两两之间的最低点作为训练的检测关键点,将其回归的loss与掌静脉检测的loss进行合并,得到最终的loss。6.如权利要求5所述的掌静脉检测模型的训练方法,其特征在于,最终的loss值L的计算方式如下:其中,λ1和λ2分别代表不同损失的权重系数,为小于1的常数;L
cls
为掌静脉分类损失函数,p
i<...

【专利技术属性】
技术研发人员:欧阳一村陈海涛龚小龙陈雅琼胡玲静陈余泉莫家源朱光强
申请(专利权)人:盛视科技股份有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1