基于多层嵌套非全映射U型网络的红外小目标检测方法
未命名
08-03
阅读:75
评论:0

基于多层嵌套非全映射u型网络的红外小目标检测方法
技术领域
1.本发明涉及计算机视觉领域中的目标检测和图像处理技术领域,特别是涉及一种基于多层嵌套非全映射u型网络的红外小目标检测方法。
背景技术:
2.红外图像是热辐射成像,它是随着红外成像技术的出现而诞生的。它是根据红外辐射的差异成像的,红外热成像系统能够把物体表面的自然发射的红外辐射分布转变为可见图像。由于红外图像是灰度图像,目标和背景的分别可以通过它们在图像中的亮度和温度区别来区分。目标通常会比背景更亮或更暗,这取决于它们的温度相对于背景的温度是更高还是更低。红外目标检测是一种不需要提供待测目标先验信息的目标检测,其通过挖掘图像自身的一些周围像素值差异来判断区域是否存在目标。红外图像的获取不依赖于外部光线,具有全天侯特点,从而使得红外图像的目标检测成为研究热点。
3.红外弱小目标检测存在以下难点:1、红外图像由于成像距离长,红外目标一般较小,图像范围从一个像素到几十个像素不等;2、红外图像的大部分区域是背景像素,目标昏暗,具有较低的信杂波比,其所占红外图像区域小,红外图像的背景和目标分布不均衡;3、由于红外弱小目标尺寸较小,表面材质和温度分布不均匀等因素,会导致红外弱小目标具有有限的形状特征;4、红外目标的大小和形状在不同的场景中差异很大,在不同的场景中,红外目标的距离和角度会发生变化,从而影响目标在红外图像中的大小和形状。
4.针对上述难点,尤其是目标在复杂的背景下所占的像素区域极小,红外图像包含多种物质(端元)的混合噪声的情况,现有方法存在对红外图像的光谱信息和空间信息挖掘不充分的问题,因此,需提供一种有效挖掘红外图像的光谱信息和空间信息的目标检测方法,以提高目标检测结果准确率。
技术实现要素:
5.为了解决以上技术问题,本发明提供一种基于多层嵌套非全映射u型网络的红外小目标检测方法,包括以下步骤s1、构建密集嵌套特征提取模型:密集嵌套特征提取模型包括密集嵌套模块、特征金字塔融合模块以及线性光谱混合模型,密集嵌套模块对数层u型网络进行堆叠,利用密集嵌套特征提取模型获取到的五个初步有效特征层进行上采样,对获取到的特征进行反卷积并对每一层的特征图像进行放大,并且进行特征融合,获得一个最终的融合了所有特征的有效特征层,再加入不同的跳跃连接;通过获取u型网络的第i层来形成密集嵌套特征提取模型,假设k
ij
表示节点的输出,其中i是沿编码器的第i个下采样层,j是沿平原跳跃路径的密集块的第j个卷积层,由k
ij
表示的特征映射堆栈公式如下所示:其中,表示同一卷积块的多个级联卷积层,表示最大池化,q
ij
表示密
集嵌套层数;l
i-1,j-1
表示i行j列层的邻层特征信息;当j》0时,每个节点接收来自密集纯跳跃连接和嵌套双向交互跳跃连接三个方向的输出;s2、构建级联模型,计算通道空间注意模块特征映射,通道空间注意模块包括两个级联注意单元,两个级联注意单元分别为通道注意力和空间注意力;s3、对特征图进行通道维度的连接,重建图像的质量和细节:以通道注意矩阵为样本,通过将两个特征图在通道维度上连接,将高分辨率特征图和低分辨率特征图进行拼接;对拼接后的特征图进行卷积操作;s4、计算全局鲁棒特征图,获取深层特征信息:在密集嵌套模块之后,通过特征金字塔融合模块来聚合所得到的多层特征,将多层特征扩展到相同大小,其中ci,h0,w0分别表示的通道数、高度以及宽度;然后,将具有空间和轮廓信息的浅层特征与具有语义信息的深层特征进行拼接,生成全局鲁棒特征图;s5、计算属于同一类像素的每一个目标的质心和目标矩阵e,根据自注意力聚合邻域信息相邻节点的特征和自身的特征来计算一个权重,然后用这个权重来加权平均邻居节点的特征;s6、根据红外图像x和重建图像xr之间的误差来提取目标,得到目标检测图像。
6.本发明进一步限定的技术方案是:进一步的,步骤s2中,对通道空间注意模块计算特征映射,k
ij
从节点由一维通道注意图和二维空间注意图进行顺序处理,其中ci表示k
ij
的通道数,表示空间维度;将r个特征映射组作为一个组进行全局平均池化,得到每个组的特征向量d,获得每个组的整体特征表示,并且利用不同的特征映射组的信息进行整合。
7.前所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,步骤s2包括以下分步骤s2.1、利用鲁棒最小单形体体积方法对线性光谱混合模型进行解混,以获得端元矩阵z;利用不同映射之间的对称比函数计算复合层之间的联通值,以获得空间注意力信息矩阵v;s2.2、通过对空间注意力信息矩阵v的激活函数sigmoid提取得到信息矩阵a,建立线性多层感知混合模型的初始目标函数,如下式所示:其中,a为信息矩阵,z为端元矩阵;s2.3、在线性多层感知混合模型的初始目标函数中引入特征因子和空间加权因子,构建通道注意力机制过程模型,如下式所示:其中,为元素级加法,为s型函数,为1
×
1卷积的线性变换;在元素级加法运
算之前,空间映射被拉伸到的大小。
8.前所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,步骤s2.3中,空间加权因子的计算包括以下步骤s2.3.1、将红外图像x分割成具有多分辨率图像x
*
,然后根据相邻像素之间的相似性进行多分辨率图像x
*
稀疏,求得权重引导丰度矩阵h;s2.3.2利用权重引导丰度矩阵h计算空间权重因子,计算过程如下式所示:其中,h(i,:)表示权重引导丰度矩阵h的第i行的元素,表示矩阵的2范数,为可调节参数,表示的第i行第j列的值。
9.前所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,步骤s2.3.1包括以下分步骤s2.3.1.1、利用stn算法将红外图像x分割为g个超像素块,构建多分辨率图像x
*
∈r
l
×b,构建基于多分辨率图像x
*
的目标函数,如下式所示:其中,和是非线性函数,它们都输出具有相同维度的向量,表示元素乘积,用于融合两个分支的信息;s2.3.1.2、采用基于变量分裂的交替方向乘子法对基于多分辨率图像x
*
的目标函数进行迭代优化求解,求得权重引导丰度矩阵h。
10.前所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,步骤s2.3中,空间权重因子的第i行第j列的第t+1次迭代结果如下式所示:其中,t表示当前迭代的次数,表示为可调节参数值,n(h)表示矩阵h中的第i行第j列元素h
ij
的邻域集合,h∈{1,2,...,u}为相邻元素集的数量,u为邻域窗口大小,h
ij
表示矩阵h中第i行第j列的值h
ij
的相邻元素集。
11.前所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,步骤s4中,通过下式计算全局鲁棒特征图g:其中,为浅层特征信息。
12.前所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,步骤s5中,通过目标矩阵e,求得与红外图像x同维度大小的重建图像xr,在特征金字塔融合模块之后,引入一个八连通邻域聚类模块,将属于同一目标的像素聚类在一起,并计算每个目标的质心;假设特征图g中的任何两个像素(m0,n0)、(m1,n1)在它们的八个邻域中都有交叉区域,如下式所示:
其中,和代表像素(m0,n0)和(m1,n1)、(m0,n0)和(m1,n1)的八个邻域被判定为相邻像素,如果这两个像素具有相同的值,该值为0或1,则认为这两个像素位于连通区域,即如下式所示;一个连接区域中的像素属于相同的目标,在确定了图像中的所有目标后,根据所有目标的坐标计算出质心,其中,g(m0,n0)和g(m1,n1)表示像素(m0,n0)和(m1,n1)的灰度值。
13.前所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,步骤s5中,自注意力聚合邻域信息结果,即通过卷积内核使用加权和聚合邻域信息的过程,如下式所示:其中,w为聚合邻域,x为加权,a
*
=i+a-[k/2],b
*
=j+b-[k/2],k是内核大小,a,b表示当前层,a
*
,b
*
表示相邻层,c表示通道。
[0014]
前所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,步骤s6包括以下分步骤s6.1、通过下式计算红外图像x和重建图像xr之间的误差:s6.2、根据误差error(x)得到灰度图像,再选取分割阈值δ,灰度图像中大于δ的像素值标记为1,表示为目标;反之标记为0,表示背景,得到只有0和1的二值结果图,得到目标检测结果。
[0015]
本发明的有益效果是:(1)本发明中,通过对多层嵌套非全映射u型网络模型进行双空间权重稀疏解混以获得丰度矩阵h,以丰度矩阵h为样本,生成字典,引入加权因子和基于自身层和相邻层之间邻域信息的空间加权因子,多层嵌套非全映射u型网络模型,以获取目标矩阵e,通过目标矩阵e,求得与原始红外弱小目标图像x同维度大小的重建图像xr,据原始图像x和重建图像xr之间的误差来提取目标;(2)本发明中,通过引入加权因子和基于空间邻域信息的空间加权因子来充分挖掘红外图像的邻域信息和像素信息的相关性,以提升红外图像目标检测检测的准确性,在真实的红外数据集上的实验结果表明,本发明获得的目标检测结果准确率为96.12%;(3)本发明中,可以有效地提取红外小目标的特征,同时抑制背景噪声和干扰;同时可以实现物体的多层次和多尺度表示学习,提高检测率和鲁棒性,降低误报率;也可以适应不同的场景和环境,如云层、雾霾、水面等。
附图说明
[0016]
图1为本发明的整体流程示意图;图2为本发明的模型结构示意图;图3为本发明实施例中实施数据集irstd-1k红外图像的真实地物位置图;图4为本发明实施例中使用method1方法的对比数据集irstd-1k红外图像的目标检测结果图;图5为本发明实施例中使用method2方法的对比数据集irstd-1k红外图像的目标检测结果图;图6为本发明实施例中实施数据集irstd-1k红外图像的目标检测结果图。
具体实施方式
[0017]
本实施例提供的一种基于多层嵌套非全映射u型网络的红外小目标检测方法,如图1所示,包括以下步骤s1、构建密集嵌套特征提取模型:如图2所示,密集嵌套特征提取模型包括密集嵌套模块、特征金字塔融合模块以及线性光谱混合模型,密集嵌套模块对数层u型网络进行堆叠,利用密集嵌套特征提取模型获取到的五个初步有效特征层进行上采样,对获取到的特征进行反卷积并对每一层的特征图像进行放大,并且进行特征融合,获得一个最终的融合了所有特征的有效特征层,再加入不同的跳跃连接;特征金字塔融合模块,可以提取和结合不同尺度的特征图,从而提高目标检测和图像分割等任务的效果;线性光谱混合模型分析红外图像,它假设每个像元的光谱是不同地物的光谱的线性组合,通过求解端元光谱和丰度系数,可以得到像元中各种地物的分布情况;通过获取u型网络的第i层来形成密集嵌套特征提取模型,假设k
ij
表示节点的输出,其中i是沿编码器的第i个下采样层,j是沿平原跳跃路径的密集块的第j个卷积层,由k
ij
表示的特征映射堆栈公式如下所示:其中,表示同一卷积块的多个级联卷积层,表示最大池化,q
ij
表示密集嵌套层数;l
i-1,j-1
表示i行j列层的邻层特征信息;当j》0时,每个节点接收来自密集纯跳跃连接和嵌套双向交互跳跃连接三个方向的输出。
[0018]
s2、构建级联模型,计算通道空间注意模块特征映射,对由两个级联注意单元组成的通道空间注意模块计算特征映射,k
ij
从节点由一维通道注意图和二维空间注意图进行顺序处理,其中ci表示k
ij
的通道数,表示空间维度;将r个特征映射组作为一个组进行全局平均池化,得到每个组的特征向量d,获得每个组的整体特征表示,并且利用不同的特征映射组的信息进行整合,提高特征的表示能力,从而提高网络的性能。
[0019]
步骤s2具体包括以下分步骤s2.1、利用鲁棒最小单形体体积方法对线性光谱混合模型进行解混,以获得端元矩阵z;利用不同映射之间的对称比函数计算复合层之间的联通值,以获得空间注意力信息
矩阵v。
[0020]
s2.2、通过对空间注意力信息矩阵v的激活函数sigmoid提取得到信息矩阵a,建立线性多层感知混合模型的初始目标函数,如下式所示:其中,a为信息矩阵,z为端元矩阵。
[0021]
s2.3、在线性多层感知混合模型的初始目标函数中引入特征因子和空间加权因子,构建通道注意力机制过程模型,如下式所示:其中,为元素级加法,为s型函数,为1
×
1卷积的线性变换;在元素级加法运算之前,空间映射被拉伸到的大小;空间权重因子的第i行第j列的第t+1次迭代结果如下式所示:其中,t表示当前迭代的次数,表示为可调节参数值,n(h)表示矩阵h中的第i行第j列元素h
ij
的邻域集合,h∈{1,2,...,u}为相邻元素集的数量,u为邻域窗口大小,h
ij
表示矩阵h中第i行第j列的值h
ij
的相邻元素集。
[0022]
步骤s2.3中,空间加权因子的计算包括以下步骤s2.3.1、将红外图像x分割成具有多分辨率图像x
*
,然后根据相邻像素之间的相似性进行多分辨率图像x
*
稀疏,具体如下:利用stn算法将红外图像x分割为g个超像素块,构建多分辨率图像x
*
∈r
l
×b,构建基于多分辨率图像x
*
的目标函数,如下式所示:其中,和是非线性函数,它们都输出具有相同维度的向量,表示元素乘积,用于融合两个分支的信息;采用基于变量分裂的交替方向乘子法对基于多分辨率图像x
*
的目标函数进行迭代优化求解,求得权重引导丰度矩阵h。
[0023]
s2.3.2利用权重引导丰度矩阵h计算空间权重因子,计算过程如下式所示:其中,h(i,:)表示权重引导丰度矩阵h的第i行的元素,表示矩阵的2范数,为可调节参数,表示的第i行第j列的值。
[0024]
s3、对特征图进行通道维度的连接,重建图像的质量和细节:以通道注意矩阵为样本,通过将两个特征图在通道维度上连接,将高分辨率特征图和低分辨率特征图进行拼接;对拼接后的特征图进行卷积操作;
以通道注意矩阵为样本,通过将两个特征图在通道维度上连接,将高分辨率特征图和低分辨率特征图进行拼接。对拼接后的特征图进行卷积操作,利用低分辨率特征图中的全局信息,以及高分辨率中局部细节信息,提高重建图像的质量和细节保留能力。
[0025]
s4、计算全局鲁棒特征图,获取深层特征信息:在密集嵌套模块之后,通过特征金字塔融合模块来聚合所得到的多层特征,将多层特征扩展到相同大小,其中ci,h0,w0分别表示的通道数、高度以及宽度;然后将具有丰富空间和轮廓信息的浅层特征与具有丰富语义信息的深层特征进行拼接,生成全局鲁棒特征图;通过下式计算全局鲁棒特征图g:其中,为浅层特征信息。
[0026]
s5、计算属于同一类像素的每一个目标的质心和目标矩阵e,根据自注意力聚合邻域信息相邻节点的特征和自身的特征来计算一个权重,然后用这个权重来加权平均邻居节点的特征;增强网络对多尺度特征的表示能力,使得模型更加灵活和可调节,提高非线性表达能力;通过目标矩阵e,求得与红外图像x同维度大小的重建图像xr,在特征金字塔融合模块之后,引入一个八连通邻域聚类模块,将属于同一目标的像素聚类在一起,并计算每个目标的质心;假设特征图g中的任何两个像素(m0,n0)、(m1,n1)在它们的八个邻域中都有交叉区域,如下式所示:其中,和代表像素(m0,n0)和(m1,n1)、(m0,n0)和(m1,n1)的八个邻域被判定为相邻像素,如果这两个像素具有相同的值,该值为0或1,则认为这两个像素位于连通区域,即如下式所示;一个连接区域中的像素属于相同的目标,在确定了图像中的所有目标后,根据所有目标的坐标计算出质心,其中,g(m0,n0)和g(m1,n1)表示像素(m0,n0)和(m1,n1)的灰度值。
[0027]
步骤s5中,自注意力聚合邻域信息结果,即通过卷积内核使用加权和聚合邻域信息的过程,如下式所示:上式看作是通过卷积内核使用加权和聚合邻域信息的过程,其中w为聚合邻域,x为加权,a
*
=i+a-[k/2],b
*
=j+b-[k/2],k是内核大小,a,b表示当前层,a
*
,b
*
表示相邻层,c表示通道。
[0028]
s6、根据红外图像x和重建图像xr之间的误差来提取目标,得到目标检测图像,具体包括以下分步骤s6.1、通过下式计算红外图像x和重建图像xr之间的误差:s6.2、根据误差error(x)得到灰度图像,再选取分割阈值δ,灰度图像中大于δ的像
素值标记为1,表示为目标;反之标记为0,表示背景,得到只有0和1的二值结果图,得到目标检测结果。
[0029]
本发明实施例采用通过数据集irstd-1k红外图像获取的户外无人机图像,irstd-ik数据集提供了1000个具有各种目标形状、不同目标大小和丰富杂波背景的真实图像,该背景具有精确的像素级注释,该数据集分为两个文件夹,分别为irstd1k_img存放真实图像和irstd1k_label存放标签mask。
[0030]
如图3所示,为irstd-1k红外图像的真实地物位置,实验区域数据大小为100
×
100,共62个像元,对应为红外图像x;分别采用mpcm、nipps、ript、ristdnet、yolov5、pstnn和本发明方法对irstd-1k红外图像进行目标检测,其目标检测准确率如表1所示,本发明的目标检测准确率高达96.12%,其明显优于其他几种方法。
[0031]
表1红外弱小目标检测准确率对比表
[0032]
假设本发明多层嵌套非全映射u型网络模型中只考虑加权因子的方法为method1,只考虑基于空间邻域信息的空间加权因子的方法为method2,如图4至图5所示,分别通过method1和method2对irstd-1k户外无人机图像进行目标检测,method1的目标检测准确率为95.86%,method2的目标检测准确率为95.89%,本发明的目标检测准确率比method1和method2分别高0.22%和0.19%,如图6所示,可以看出本发明的目标检测效果明显要比method1和method2好。
[0033]
除上述实施例外,本发明还可以有其他实施方式。凡采用等同替换或等效变换形成的技术方案,均落在本发明要求的保护范围。
技术特征:
1.一种基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:包括以下步骤s1、构建密集嵌套特征提取模型:密集嵌套特征提取模型包括密集嵌套模块、特征金字塔融合模块以及线性光谱混合模型,密集嵌套模块对数层u型网络进行堆叠,利用密集嵌套特征提取模型获取到的五个初步有效特征层进行上采样,对获取到的特征进行反卷积并对每一层的特征图像进行放大,并且进行特征融合,获得一个最终的融合了所有特征的有效特征层,再加入不同的跳跃连接;通过获取u型网络的第i层来形成密集嵌套特征提取模型,假设k
ij
表示节点 的输出,其中i是沿编码器的第i个下采样层,j是沿平原跳跃路径的密集块的第j个卷积层,由k
ij
表示的特征映射堆栈公式如下所示:其中,表示同一卷积块的多个级联卷积层,表示最大池化,q
ij
表示密集嵌套层数;l
i-1,j-1
表示i行j列层的邻层特征信息;当j>0时,每个节点接收来自密集纯跳跃连接和嵌套双向交互跳跃连接三个方向的输出;s2、构建级联模型,计算通道空间注意模块特征映射,通道空间注意模块包括两个级联注意单元,两个级联注意单元分别为通道注意力和空间注意力;s3、对特征图进行通道维度的连接,重建图像的质量和细节:以通道注意矩阵为样本,通过将两个特征图在通道维度上连接,将高分辨率特征图和低分辨率特征图进行拼接;对拼接后的特征图进行卷积操作;s4、计算全局鲁棒特征图,获取深层特征信息:在密集嵌套模块之后,通过特征金字塔融合模块来聚合所得到的多层特征,将多层特征扩展到相同大小,其中c
i
,h0,w0分别表示的通道数、高度以及宽度;然后,将具有空间和轮廓信息的浅层特征与具有语义信息的深层特征进行拼接,生成全局鲁棒特征图;s5、计算属于同一类像素的每一个目标的质心和目标矩阵e,根据自注意力聚合邻域信息相邻节点的特征和自身的特征来计算一个权重,然后用这个权重来加权平均邻居节点的特征;s6、根据红外图像x和重建图像x
r
之间的误差来提取目标,得到目标检测图像。2.根据权利要求1所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:所述步骤s2中,对通道空间注意模块计算特征映射,k
ij
从节点由一维通道注意图和二维空间注意图进行顺序处理,其中c
i
表示k
ij
的通道数,表示空间维度;将r个特征映射组作为一个组进行全局平均池化,得到每个组的特征向量d,获得每个组的整体特征表示,并且利用不同的特征映射组的信息进行整合。3.根据权利要求1所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特
征在于:所述步骤s2包括以下分步骤s2.1、利用鲁棒最小单形体体积方法对线性光谱混合模型进行解混,以获得端元矩阵z;利用不同映射之间的对称比函数计算复合层之间的联通值,以获得空间注意力信息矩阵v;s2.2、通过对空间注意力信息矩阵v的激活函数sigmoid提取得到信息矩阵a,建立线性多层感知混合模型的初始目标函数,如下式所示:其中,a为信息矩阵,z为端元矩阵;s2.3、在线性多层感知混合模型的初始目标函数中引入特征因子和空间加权因子,构建通道注意力机制过程模型,如下式所示:其中,为元素级加法,为s型函数,为1
×
1卷积的线性变换;在元素级加法运算之前,空间映射被拉伸到的大小。4.根据权利要求3所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:所述步骤s2.3中,空间加权因子的计算包括以下步骤s2.3.1、将红外图像x分割成具有多分辨率图像x
*
,然后根据相邻像素之间的相似性进行多分辨率图像x
*
稀疏,求得权重引导丰度矩阵h;s2.3.2利用权重引导丰度矩阵h计算空间权重因子,计算过程如下式所示:其中,h(i,:)表示权重引导丰度矩阵h的第i行的元素,表示矩阵的2范数,为可调节参数,表示的第i行第j列的值。5.根据权利要求4所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:所述步骤s2.3.1包括以下分步骤s2.3.1.1、利用stn算法将红外图像x分割为g个超像素块,构建多分辨率图像x
*
∈r
l
×
b
,构建基于多分辨率图像x
*
的目标函数,如下式所示:其中,和是非线性函数,它们都输出具有相同维度的向量,表示元素乘积,用于融合两个分支的信息;s2.3.1.2、采用基于变量分裂的交替方向乘子法对基于多分辨率图像x
*
的目标函数进行迭代优化求解,求得权重引导丰度矩阵h。6.根据权利要求3所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:所述步骤s2.3中,空间权重因子的第i行第j列的第t+1次迭代结果如下式所示:
其中,t表示当前迭代的次数,表示为可调节参数值,n(h)表示矩阵h中的第i行第j列元素h
ij
的邻域集合,h∈{1,2,...,u}为相邻元素集的数量,u为邻域窗口大小,h
ij
表示矩阵h中第i行第j列的值h
ij
的相邻元素集。7.根据权利要求1所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:所述步骤s4中,通过下式计算全局鲁棒特征图g:其中,为浅层特征信息。8.根据权利要求1所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:所述步骤s5中,通过目标矩阵e,求得与红外图像x同维度大小的重建图像x
r
,在特征金字塔融合模块之后,引入一个八连通邻域聚类模块,将属于同一目标的像素聚类在一起,并计算每个目标的质心;假设特征图g中的任何两个像素(m0,n0)、(m1,n1)在它们的八个邻域中都有交叉区域,如下式所示:其中,和代表像素(m0,n0)和(m1,n1)、(m0,n0)和(m1,n1)的八个邻域被判定为相邻像素,如果这两个像素具有相同的值,该值为0或1,则认为这两个像素位于连通区域,即如下式所示;一个连接区域中的像素属于相同的目标,在确定了图像中的所有目标后,根据所有目标的坐标计算出质心,其中,g(m0,n0)和g(m1,n1)表示像素(m0,n0)和(m1,n1)的灰度值。9.根据权利要求1所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:所述步骤s5中,自注意力聚合邻域信息结果,即通过卷积内核使用加权和聚合邻域信息的过程,如下式所示:其中,w为聚合邻域,x为加权,a
*
=i+a-[k/2],b
*
=j+b-[k/2],k是内核大小,a,b表示当前层,a
*
,b
*
表示相邻层,c表示通道。10.根据权利要求1所述的基于多层嵌套非全映射u型网络的红外小目标检测方法,其特征在于:所述步骤s6包括以下分步骤s6.1、通过下式计算红外图像x和重建图像x
r
之间的误差:s6.2、根据误差error(x)得到灰度图像,再选取分割阈值δ,灰度图像中大于δ的像素值标记为1,表示为目标;反之标记为0,表示背景,得到只有0和1的二值结果图,得到目标检测结果。
技术总结
本发明公开了一种基于多层嵌套非全映射U型网络的红外小目标检测方法,涉及目标检测和图像处理技术领域,包括以下步骤:步骤1、构建密集嵌套特征提取模型;步骤2、构建级联模型,计算通道空间注意模块特征映射;步骤3、对特征图进行通道维度的连接,重建图像的质量和细节;步骤4、计算全局鲁棒特征图,获取深层特征信息;步骤5、计算属于同一类像素的每一个目标的质心;步骤6、根据红外图像和重建图像得到目标检测图像;从而通过引入级联模型和计算全局鲁棒特征图来充分挖掘红外图像的邻域信息和空间信息的相关性,以提升红外图像目标检测的准确性,本发明的目标检测结果准确率为96.12%,相比现有技术具有更高的准确性。相比现有技术具有更高的准确性。相比现有技术具有更高的准确性。
技术研发人员:涂兵 沈文增 李武劲 郭龙源 何伟
受保护的技术使用者:南京信息工程大学
技术研发日:2023.06.29
技术公布日:2023/8/1
版权声明
本文仅代表作者观点,不代表航家之家立场。
本文系作者授权航家号发表,未经原创作者书面授权,任何单位或个人不得引用、复制、转载、摘编、链接或以其他任何方式复制发表。任何单位或个人在获得书面授权使用航空之家内容时,须注明作者及来源 “航空之家”。如非法使用航空之家的部分或全部内容的,航空之家将依法追究其法律责任。(航空之家官方QQ:2926969996)
航空之家 https://www.aerohome.com.cn/
飞机超市 https://mall.aerohome.com.cn/
航空资讯 https://news.aerohome.com.cn/