一种基于动态卷积的遥感图像舰船目标细粒度分类方法技术

本发明专利技术公开了一种基于动态卷积的遥感图像舰船目标细粒度分类方法，将采集特征图输入到注意力模块中，注意力模块生成K个归一化注意力权重参数；对采集特征图卷积处理中，平行卷积核模块中用K个平行卷积代替单独卷积，注意力权重参数与K个平行卷积的卷积核进行结合形成动态卷积层，最后连接到分类网络中进行分类，本发明专利技术通过动态卷积层与分类网络的结合，将注意力机制融入动态卷积中，多核集成并且添加了经过前面注意力模块计算归一化的注意力权重参数以非线性的方式融合，提高了模型的特征学习能力进而提高舰船目标细粒度分类的准确率。确率。确率。

【技术实现步骤摘要】
一种基于动态卷积的遥感图像舰船目标细粒度分类方法


[0001]本专利技术属于数字图像处理领域，更具体的说是涉及一种基于动态卷积的遥感图像舰船目标细粒度分类方法。

技术介绍

[0002]遥感图像海上舰船目标的分类技术一直是遥感图像处理领域的基础研究，在港口实时监控、海上军事舰船目标检测和精细识别方面具有重要的意义。由于公开的大部分有关遥感图像舰船目标研究主要集中舰船目标的检测和后续进行背景和舰船目标的二分类，少部分的文献专门对舰船目标多分类任务进行了研究，但是分类的类别数量也十分有限，近年来，深度学习技术的快速发展，细粒度分类相关的技术受到了越来越多的关注，细粒度分类技术可以舰船目标进行更精细化的分类。
[0003]目前图像细粒度分类主要的注意力机制都是和特征图通道层面进行操作，也就是将注意力机制与传统的卷积进行融合，传统卷积相当于是静态卷积其可以用公式表示，y＝g(W
T
(x)+b(x))，静态卷积运算的卷积核是空间共享的，经过静态的卷积运算得到的特征图较为平滑，容易丢失原始图像中的信息，因此将基于卷积层的注意力的动态卷积融合到细粒度分类中是一个重要的应用体现，将对舰船目标细粒度分类的准确率起到提高的作用。
[0004]因此，为了解决遥感图像舰船目标细粒度分类准确率低的问题，提出了一种基于动态卷积的遥感图像舰船目标细粒度分类技术。

技术实现思路

[0005]有鉴于此，本专利技术提供了一种基于动态卷积的遥感图像舰船目标细粒度分类方法，通过的动态卷积层使注意力函数对输入图像产生依赖，具有更强特征提取能力进而提高了分类准确率。
[0006]为了实现上述目的，本专利技术采用如下技术方案：
[0007]一种基于动态卷积的遥感图像舰船目标细粒度分类方法，将采集特征图输入到注意力模块中，注意力模块生成K个归一化注意力权重参数；
[0008]对采集特征图卷积处理中，平行卷积核模块中用K个平行卷积代替单独卷积，注意力权重参数与K个平行卷积的卷积核进行结合形成动态卷积层，最后连接到分类网络中进行分类。
[0009]进一步，所述注意力模块建模方法如下，
[0010]S1：采集动态输入的特征图，通过池化操作捕捉特征图信息得到池化特征图，池化特征图维度为1
×1×
c，c为通道数；
[0011]S2：池化特征图进入第一全连接层进行降维，得到第一特征图；
[0012]S3：通过ReLU函数对降维后的第一特征图引入非线性因素，克服梯度消失问题，得到第二特征图；
[0013]S4：第二特征图进入第二全连接层进行升维，维度升至1
×1×
c得到第三特征图，
即可以表示为元素个数为c的一维向量；
[0014]S5：利用softmax函数对一维向量进行归一化，得到K个注意力权重。
[0015]进一步，所述S1中的池化操作为对输入的特征图分别进行平均池化和最大池化，两种池化下的池化特征图平行运行，在所述步骤S5中将平均池化下的第三特征图和最大池化下的第三特征图分别对应一个一维向量，两个一维向量对应相加后通过softmax函数归一化，得到K个注意力权重参数。
[0016]进一步，所述平行卷积核模块为动态输入的特征图对应的K个平行卷积核，所述K个平行卷积核与K个注意力权重对应相乘后相加得到新卷积核，所述新的卷积核以卷积的方式结合到分类网络中。
[0017]进一步，所述动态卷积层采用Block结构设计，方便嵌入到各种分类网络中进行分类。
[0018]进一步，所述分类网络为VGG或ResNet。
[0019]本专利技术的技术效果：
[0020]本专利技术采用的基于动态卷积的舰船目标细粒度分类算法，首先将注意力机制融合到卷积层层面上进行操作，动态卷积根据输入特征图集成多个平行的卷积核构成一个动态卷积层，利用原本就是这个卷积层会随着输入的改变而变化，多核集成并且添加了经过前面注意力模块计算归一化的注意力权重参数以非线性的方式融合，这样提高了模型的特征学习能力；另外结合最大池化和平均池化的注意力获取方式可以更有效的发挥注意力机制在卷积层上的作用，结合相应的分类网络，本专利技术提高舰船目标细粒度分类的准确率。
附图说明
[0021]为了更清楚地说明本专利技术实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本专利技术的实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据提供的附图获得其他的附图。
[0022]图1为本专利技术动态卷积网络示意图；
[0023]图2为本专利技术动态卷积原理图。
具体实施方式
[0024]下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。
[0025]如图1，图2一种基于动态卷积的遥感图像舰船目标细粒度分类方法，将采集特征图输入到注意力模块中，注意力模块生成K个归一化注意力权重参数；
[0026]对采集特征图卷积处理中，平行卷积核模块中用K个平行卷积代替单独卷积，注意力权重参数与K个平行卷积的卷积核进行结合形成动态卷积层，最后连接到分类网络中进行分类。
[0027]其中，与传统的卷积方式不的是，传统的卷积相当于是一种静态的卷积方式，动态
卷积相当于将不同的权重加到平行的卷积核上，其动态卷积模型表达式为，
[0028][0029][0030]公式中分别是个K网络的卷积核参数和偏置参数，根据权重函数π(x)将K个不同个卷积核进行对应元素进行求和，能够得到处理后的卷积核参数和偏置参数W和b；π(x)为作为权重函数，作为一个非线性的函数，这样动态卷积各平行卷积核以非线性的方式融合就能够具有更强的表达能力。
[0031]为了进一步优化上述技术方案，注意力模块建模方法如下，
[0032]S1：采集动态输入的特征图x，通过池化操作捕捉特征图信息得到池化特征图，池化特征图维度为1
×1×
c，c为通道数；
[0033]S2：池化特征图进入第一全连接层进行降维，得到第一特征图；
[0034]S3：通过ReLU函数对降维后的第一特征图引入非线性因素，克服梯度消失问题，得到第二特征图；
[0035]S4：第二特征图进入第二全连接层进行升维，维度升至1
×1×
c得到第三特征图，即可以表示为元素个数为c的一维向量；
[0036]S5：利用softmax函数对一维向量进行归一化，得到K个注意力权重。
[0037]为了进一步优化上述技术方案，步骤S1中的池化操作为对输入的特征图分别进行平均池化和最大池化，两种池化下的池化特征图平行运行，在步骤S5中将平均池化下的第本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于动态卷积的遥感图像舰船目标细粒度分类方法，其特征在于，将采集特征图输入到注意力模块中，注意力模块生成K个归一化注意力权重参数；对采集特征图卷积处理中，平行卷积核模块中用K个平行卷积代替单独卷积，注意力权重与K个平行卷积的卷积核进行结合形成动态卷积层，最后连接到分类网络中进行分类。2.根据权利要求1中所述的一种基于动态卷积的遥感图像舰船目标细粒度分类方法，其特征在于，所述注意力模块建模方法如下，S1：采集动态输入的特征图，通过池化操作捕捉特征图信息得到池化特征图，池化特征图维度为1
×1×
c，c为通道数；S2：池化特征图进入第一全连接层进行降维，得到第一特征图；S3：通过ReLU函数对降维后的第一特征图引入非线性因素，克服梯度消失问题，得到第二特征图；S4：第二特征图进入第二全连接层进行升维，维度升至1
×1×
c得到第三特征图，即可以表示为元素个数为c的一维向量；S5：利用softmax函数对一维向量进行归一化，得到K个注意力...

【专利技术属性】
技术研发人员：姜志国，邸杨骅，张浩鹏，赵丹培，谢凤英，
申请(专利权)人：北京航空航天大学，
类型：发明
国别省市：