一种基于深度极化哈希的分层车辆图像检索方法与流程

文档序号:32058404发布日期:2022-11-04 22:21阅读:来源:国知局

技术特征:
1.一种基于深度极化哈希的分层车辆图像检索方法,其特征在于,包括:获取车辆图像数据集并进行预处理;搭建分层极化网络模型,进行极化损失函数和优化器的配置,并导入车辆图像数据集进行特征学习和哈希函数学习的训练;利用训练好的分层极化网络模型提取数据集中查询图像和数据库图像的超类特征和子类特征,并利用学习后的哈希函数对超类特征和子类特征进行哈希化,得到哈希码;通过得到的哈希码的超类哈希特征和哈希子类特征计算相似度进行分层图像检索,得到检索结果。2.根据权利要求1所述一种基于深度极化哈希的分层车辆图像检索方法,其特征在于,所述获取车辆图像数据集并进行预处理的具体步骤包括:获取车辆图像数据集,所述车辆图像数据集包括m张图像的训练集和n张图像的测试集,其中将m张图像的训练集作为待检索数据库;对车辆图像训练集进行预处理,根据车辆品牌划分为若干个超类,再在每个超类品牌中根据不同车辆型号划分为若干个子类;对输入的车辆图像训练集的rbg通道归一化处理,限定于-1至1之间,并采用裁剪、翻转,以及旋转的数据增强方法对车辆图像固定大小尺寸。3.根据权利要求1所述一种基于深度极化哈希的分层车辆图像检索方法,其特征在于,所述搭建分层极化网络模型,进行极化损失函数和优化器的配置,并导入车辆图像数据集进行特征学习和哈希函数学习的训练的具体步骤包括:获取车辆图像数据集的训练集和测试集;搭建分层极化网络模型,制定深度模型训练策略,其中训练集与测试集的比例为m:n=4:1;再对输入的车辆图像样本进行归一化处理:(image/255)
×
2-1;其中,image为输入的车辆图像样本,采用数据增强方法扩大训练样本量,对车辆图像统一调整尺寸大小到256*256,然后剪裁尺寸大小到224*224;通过模型的骨干网络学习深度特征,对划分为超类和子类分支学习对应的超类特征和子类特征;将模型的优化器设定为adam,采用可微分的极化损失函数优化网络参数,设置哈希函数的初始化长度为bit位的{1,-1}随机值,然后采用自适应更新的方式更新超类和子类对应的哈希函数。4.根据权利要求3所述一种基于深度极化哈希的分层车辆图像检索方法,其特征在于,所述通过模型的骨干网络学习深度特征,对划分为超类和子类分支学习对应的超类特征和子类特征的具体步骤包括:将resnet50网络作为骨干网络提取每张图片的深度特征,公式为:x=resnet(t);然后设置超类分支为一个全连接层网络学习超类特征v
sup
=fc(x),子类分支为d个全连接层网络学习每一个超类的子类特征v
isub
=fc(x),i∈{1,2,

,d};其中,resnet()为resnet50网络,fc()为全连接层网络,d为超类类别数。5.根据权利要求3所述一种基于深度极化哈希的分层车辆图像检索方法,其特征在于,所述采用可微分的极化损失函数优化网络参数的具体步骤包括:
采用可微分的极化损失函数使得骨干网络的输出特征v中的每个元素均偏离0,所述极化损失函数的公式:其中,v为每个分支的超类子类特征,t为要学习的哈希函数,m为阈值,用于控制极化损失的极化程度,k为哈希位数bit;得到目标函数为:其中,w为网络中要学习的权重,l为类别数,c
x
为第x类的样本集;将得到的超类损失l
psup
和d个子类损失l
pisub
之和作为最终的总损失,i∈{1,2,

,d}。6.根据权利要求1所述一种基于深度极化哈希的分层车辆图像检索方法,其特征在于,所述采用自适应更新的方式更新超类和子类对应的哈希函数的具体步骤包括:对于每一个类别c,获取其预测正确的特征v
c
,然后将该特征值符号化作为更新后的哈希值,即t
c_new
=sign(v
c
),其中sign()为符号函数。7.根据权利要求1所述一种基于深度极化哈希的分层车辆图像检索方法,其特征在于,所述利用学习后的哈希函数对超类特征和子类特征进行哈希化的具体步骤包括:获取数据集中的查询图像和数据库图像,提取对应的超类特征和子类特征;对得到的超类特征和子类特征,采用三元赋值的方式和学习后的哈希函数进行哈希化,得到最终的哈希码,公式为:其中,x为超类和子类特征的各个元素的特征值,m为极化损失中的阈值。8.根据权利要求1所述一种基于深度极化哈希的分层车辆图像检索方法,其特征在于,所述通过得到的哈希码的超类哈希特征和哈希子类特征计算相似度进行分层图像检索,得到检索结果的具体步骤包括:利用数据集中的查询图像和数据库图像的超类哈希特征计算之间的汉明距离,得到一个候选排序结果,汉明距离的计算公式为:其中,b和t分别代表要计算的两个向量,k代表哈希位数bit;获取候选排序结果中的前k个数据库图像进行子类特征相似度计算,其中k为数据库图像数量的1/10;最后计算子类哈希特征之间的汉明距离得到最终的检索排序结果。

技术总结
本发明公开了一种基于深度极化哈希的分层车辆图像检索方法,包括获取车辆图像数据集并进行预处理;搭建分层极化网络模型,进行极化损失函数和优化器的配置,并导入车辆图像数据集进行特征学习和哈希函数学习的训练;利用训练好的分层极化网络模型提取数据集中查询图像和数据库图像的超类特征和子类特征,并利用学习后的哈希函数对超类特征和子类特征进行哈希化,得到哈希码;通过得到的哈希码的超类哈希特征和哈希子类特征计算相似度进行分层图像检索,得到检索结果。本发明通过利用车辆图片的超类子类标签学习超类子类哈希特征,采用分层哈希检索的方式,检索出与查询图片同一型号的车辆,加快车辆图像检索的速度和精度。度。度。


技术研发人员:王雪雁 陈晨 刘晓静 王新朋 秦玉莉
受保护的技术使用者:合肥市正茂科技有限公司
技术研发日:2022.08.24
技术公布日:2022/11/3
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1