【技术实现步骤摘要】
一种基于优化自适应度量学习的图像匹配方法及系统
[0001]本专利技术属于人工智能
,涉及一种图像匹配方法及系统,具体涉及一种基于优化自适应度量学习的图像匹配方法及系统。
技术介绍
[0002]图像匹配是人工智能领域中重要的研究范畴,包括人脸识别,图像检索,行人重识别等在内的众多经典模式识别问题,都可以归结为图像匹配问题。当前大部分的图像匹配方法都强依赖于深度学习框架,而训练策略和损失函数时深度学习中最为重要的内容之一,其设计的合理性和准确性将直接影响到图像的深度特征表达,进而影响图像匹配的最终结果。度量学习是图像匹配中常用的损失函数构建方法,其方法核心是在样本特征空间中,以增加类间样本间距离和缩短类内样本间距离为主要优化方向,实现样本从原始图像空间到特征度量空间的映射,其中三元组损失函数时度量学习中应用最广泛的损失函数之一。此方法能够一定程度上弥补表征学习(Representation Learning)对样本特征表达欠准确的问题,但仍然存在以下两个问题:
[0003]1)低效的优化过程。在反向传播时,三元组 ...
【技术保护点】
【技术特征摘要】
1.一种基于优化自适应度量学习的图像匹配方法,其特征在于,包括以下步骤:步骤1:构建优化自适应度量学习网络;所述优化自适应度量学习网络,是在Dnet最后的输出层添加一个输出维度为K的全连接层将输出结果变换为1
×
K的向量作为图像的特征向量f
pre
,在此基础上再添加一个输出维度为L的全连接层,输出1
×
L的输出结果向量V
p
;步骤2:训练所述优化自适应度量学习网络,获得训练好的优化自适应度量学习网络;其中,训练样本三元组记为其中,锚点样本记为正样本为与锚点标签相同的样本数据记为负样本为与锚点标签不同的样本数据记为步骤3:将需要匹配图像输入所述训练好的优化自适应度量学习网络,输出匹配图像。2.根据权利要求1所述的基于优化自适应度量学习的图像匹配方法,其特征在于,步骤2的具体实现包括以下子步骤:步骤2.1:构建图像训练样本集;已知图像训练样本集P
i
=(img
i
,ID
i
),其中P
i
表示训练集中第i个图像样本,ID
i
∈N
L
是该样本的标签向量,L是样本集中所有样本类别的总数,N是样本总数目;img
i
表示P
i
对应的样本图像;步骤2.2:将图像训练样本集中样本i输入所述优化自适应度量学习网络,输出1
×
L结果向量步骤2.3:利用Softmax公式将归一化为向量归一化为向量步骤2.4:利用公式(2)所示的交叉熵损失函数,通过反向传播的方式对网络参数进行优化;对网络进行预设轮次的循环训练后,得到训练好的优化自适应度量学习网络A;其中y
il
表示第i个样本的真实标签向量中第l维的向量值,是第i样本的输出标签向量第l维的向量值。3.根据权利要求2所述的基于优化自适应度量学习的图像匹配方法,其特征在于:对优化自适应度量学习网络A进行再训练,获得训练好的优化自适应度量学习网络;具体实现包括以下子步骤:步骤2.5:将图像训练样本集中样本i输入所述优化自适应度量学习网络A,输出待匹配图像的特征向量集合记为将样本从图像空间映射至初始特征空间;步骤2.6:使用K
‑
Means聚类方法对进行聚类,使用欧氏距离度量任意两点之间的距离,预设置聚类的类别为W;随机初始化W个样本点,作为W个类簇的类中心,计算若干数据点到中心点的距离,数据点的类别被定义为A
i
当且仅当该数据点与该类别的中心点小于其到其他中心点的距离;利用多维空间类中心计算公式计算新的类别的中心点,循转执行直至每一类类中心不再变化为止,得到聚类结果;
步骤2.7:根据步骤2.6的聚类结果,为每个待匹配图像...
【专利技术属性】
技术研发人员:何政,朱玟谦,叶刚,王中元,傅佑铭,邹勤,
申请(专利权)人:武汉大学,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。