一种基于全局注意力关系网络的小样本船舶目标识别方法技术

技术编号:31709626 阅读:29 留言:0更新日期:2022-01-01 11:12
本发明专利技术提供一种基于全局注意力关系网络的小样本船舶目标识别方法,包括以下步骤:S1:使用Meta

【技术实现步骤摘要】
一种基于全局注意力关系网络的小样本船舶目标识别方法


[0001]本专利技术涉及深度学习与计算机视觉领域,具体为一种基于全局注意力关系网络的小样本船舶目标识别方法。

技术介绍

[0002]小样本目标识别的方法主要有两种策略:有监督情节式学习和无监督表征式学习。对于有监督情节式学习,是通过每个任务支持集(support)的少标签样本信息与查询集(query)样本进行特征比较,度量support和query中样本之间的距离,通过“参数更新”和“学习比较”,使学习适应新任务,旨在学习基于特征空间和分类识别的良好特征。对于无监督表征式学习,训练阶段利用无标注的数据进行学习,元测试阶段,有监督和无监督的设置相同,在支持集和查询集中均有部分辅助标注标签。为了进一步减少整体的标签学习过程,在元训练期间没有任何标签,降低数据的标签成本。除上述方法外,现在研究的表示学习与图神经网络(GNN)联系较为密切,大多数现有的图神经网络模型依赖于给定图的拓扑结构。在执行图卷积过程中,在给定的图形中直接从邻近聚合阶段特征,并堆叠多个GNN层以捕获远程节点的依赖性。上述方法,模本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于全局注意力关系网络的小样本船舶目标识别方法,其特征是:包括以下步骤:S1:使用Meta

Learning数据划分方式生成小样本任务集,并进行模型加载及训练;S2:搭建跨目标全局注意力机制模型;S3:搭建基于跨目标全局注意力机制的RN网络结构;S4:使用Cosine距离度量样本特征相似度,并对距离长度进行比较目标样本特征;S5:使用训练优化方法训练网络。2.根据权利要求1所述的一种基于全局注意力关系网络的小样本船舶目标识别方法,其特征是:所述步骤S1包括以下步骤:S1.1:将数据集训练划分为三个数据集:训练集,支持集和测试集。支持集和测试集共享相同的标签空间,但是训练集具有自己的标签空间,该标签空间与支持/测试集不相交;S1.2:训练时在每次训练迭代中,通过从训练集中随机选择C个类别(其中每个类别有K个标记样本)来形成每个训练任务;S1.3:每个训练任务的样本集合分为:支持集:S={(x
i
,y
i
)},m
i
=1,m=k
×
C;查询集:Q={(x
i
,y
i
)},n
j=1
;式中,C类样本的其余部分为未标记的样本。3.根据权利要求2所述的一种基于全局注意力关系网络的小样本船舶目标识别方法,其特征是:所述步骤S2包括以下步骤:S2.1:对特征提取f进行两支路卷积操作;S2.2:对两支路卷积后特征相加后进行softmax平滑:N=softmax(a+b);式中,a为一支路输出,b为二支路输出;S2.3:对平滑后的特征图像与特征提取f进行相减后传入FC层进行融合:I=Wg*f+bg;式中f∈R
w
×
h
×
C
,w、h、c分别表示宽度、高度和通道数,A={α
i
∈R
c
,i=1,...,D},w
g
∈R1×1×
C
×
D
和bg∈R
D
是卷积的参数。V∈R
D
×
C
表示输出描述符。f
j

α
i
表示残差操作。4.根据权利要求3所述的一种基于全局注意力关系网络的小样本船舶目标识别方法,其特征是:所述步骤S3包括以下步骤:S3.1:将每个训练任务传入到卷积神经网络Conv Block(4)中,输出特征提取f;S3.2:从Conv Block(4)中的f提取两支路特征作...

【专利技术属性】
技术研发人员:孟浩田洋李涛孙宇婷孔令儒
申请(专利权)人:哈尔滨工程大学
类型:发明
国别省市:

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

1