一种基于双子网络的红外弱小目标检测方法技术

本发明专利技术公开了一种基于双子网络的红外弱小目标检测方法，包括如下步骤：步骤1，分别构建包含两个子网络的主体框架；步骤2，对训练图像进行数据预处理送入网络；步骤3，针对子网络1，计算漏检抑制损失函数，并更新参数；针对子网络2，计算虚警抑制损失函数，并更新参数；步骤4，转至步骤2，直至训练收敛，保存两个子网络；步骤5，在测试阶段，把图像分别输入两个子网络，得到两组结果；步骤6，将两组检测结果融合，输出最终结果。本申请针对红外弱小目标图像特点，设计了一种基于双子网络的红外弱小目标检测方法，使用两个子网络分别针对红外弱小目标检测的虚警率和漏检率的任务，进一步提高红外弱小目标检测精度和鲁棒性。红外弱小目标检测精度和鲁棒性。红外弱小目标检测精度和鲁棒性。

【技术实现步骤摘要】
一种基于双子网络的红外弱小目标检测方法


[0001]本专利技术属于图像处理与模式识别
，具体涉及到一种基于双子网络的红外弱小目标检测方法，可以应用到军事预警、医学成像病理分析、海上救援等诸多领域。

技术介绍

[0002]在实际的红外探测系统中，红外图像中的弱小目标检测技术在军事领域已成为成像制导、告警系统的核心技术。由于它只对目标的温度和本身的材料特性敏感，受环境的影响小，使得它在主流的探测技术中脱颖而出，成为传统可见光探测系统与雷达探测系统的有效补充或替代，所以检测的准确性和鲁棒性至关重要，检测方法直接关乎应用系统的性能，继而影响后继的跟踪识别精度。因此研究自动的、快速的、高效的红外弱小目标检测算法对红外图像进行处理迫在眉睫。
[0003]传统红外弱小目标检测算法背景预测的红外图像处理算法对红外背景的泛化能力较差，对于背景复杂、干扰较多的场景检测性能较差，而深度神经网络在人工智能领域已经有了蓬勃的发展，通用目标检测网络和语义分割网络应用到了多个领域。研究人员针对红外检测任务有了多种不同的尝试，相关工作有：娄康提出了一种结合深度学习和生物视觉机制的红外目标检测算法，构建生成式对抗神经网络，训练生成器产生红外图像样本，并通过Gabor核函数模拟生物视觉神经单元响应对红外图像降噪处理(见参考文献：基于深度学习与生物视觉机制的红外弱小目标检测与跟踪方法研究[D].江苏科技大学,2020)；黄乐弘根据空中红外目标图像像素分辨率较低、图像边缘模糊和对比性较差的特点，对YOLO v3模型进行了改进，从提高检测定位精度的角度改进了边缘损失函数(见参考文献：黄乐弘.基于深度学习的空中红外目标检测方法研究[J].中国科学院大学,2020)；杨其利[27]等人提出了一种以语义分割为基础，采用全卷积递归网络的弱小目标检测方法，通过在网络中添加残差模块来学习复杂背景下弱小目标的特征，核心思想是加入了递归操作等等(见参考文献：杨其利,周炳红,郑伟.基于全卷积递归网络的弱小目标检测方法[J].光学学报,2020,40(13):1
‑
13)。

技术实现思路

[0004]专利技术目的：针对通用目标检测网络和语义分割网络直接迁移到红外小目标检测任务中的性能大幅下降问题，本专利技术提供一种基于双子网络的红外弱小目标检测方法。
[0005]技术方案：针对红外弱小目标图像特点，本专利技术提供一种基于双子网络的红外弱小目标检测方法。该双子网络包含两个子网络分别针对红外弱小目标检测的虚警率和漏检率的任务，最后输出层进行融合分割结果，进一步提高红外弱小目标检测精度和鲁棒性。
[0006]该方法包括如下步骤：
[0007]步骤1，分别构建子网络1和子网络2的主体框架；
[0008]步骤2，输入一批训练图像，经过数据预处理，送入到网络中；
[0009]步骤3，针对子网络1，计算漏检抑制损失函数，更新子网络1参数；针对子网络2，计
算虚警抑制损失函数，更新子网络2参数；
[0010]步骤4，转至步骤2，直至训练收敛，得到最终的子网络1和子网络2；
[0011]步骤5，在测试阶段，把测试图像分别输入最终的子网络1和子网络2，得到两组检测结果；
[0012]步骤6，将来自于子网络1和子网络2的结果进行融合，输出最终检测结果。
[0013]优选地，所述步骤1需要构造子网络1和子网络2两种网络结构，二者的区别在于：网络1旨在降低图像分割的漏检率，即学习较低的阈值，使得网络的召回率升高；网络2旨在降低图像的虚警率，即学习较高的阈值，导致网络的精确率升高。
[0014]优选地，所述步骤2中，首先导入3通道的原始输入图像，将其调整尺寸到同一大小，并转换为单通道灰度图像送入子网络1和子网络2中。
[0015]优选地，所述步骤3中，为了降低子网络1对目标的漏检率，基于SmoothL1Loss，设计了一种漏检抑制的损失函数，对漏检导致的损失具有更大的权重。
[0016]优选地，所述步骤3中，为了降低子网络2对目标的虚警率，基于SmoothL1Loss，设计了一种虚警抑制的损失函数，对虚警导致的损失具有更大的权重。
[0017]优选地，所述步骤4中，分别训练子网络1和2，保存收敛后的权重文件。
[0018]优选地，所述步骤5中，将同样的测试图像分别输入到子网络1和2中，得到同一测试图像在不同模型下的输出。
[0019]优选地，所述步骤6中，为了统一地逼近真实结果，设计了一种加权融合的形式来合并子网络1和子网络2的输出，得到最终的检测结果。
[0020]与现有技术相比，本专利技术的有益效果在于：针对红外弱小目标图像特点，设计了一种基于双子网络的红外弱小目标检测方法；包含两个子网络分别针对红外弱小目标检测的虚警率和漏检率的任务，最后输出层进行融合分割结果，有效提高了红外弱小目标检测精度和鲁棒性。
附图说明
[0021]图1为本专利技术所提供的基于双子网络的红外弱小目标检测方法的流程示意图。
[0022]图2为本专利技术所构建的子网络1架构示意图。
[0023]图3为本专利技术所构建的子网络2架构示意图。
[0024]图4为本专利技术所采用的训练样本示意图。
[0025]图5为本专利技术所提供的基于双子网络的红外弱小目标检测方法的检测结果可视化。
[0026]图6为本专利技术所构建的子网络1和子网络2的损失值随迭代次数变化图。
具体实施方式
[0027]下面参照附图来描述本专利技术的优选实施方式。本领域技术人员应当理解的是，这些实施方式仅仅用于解释本专利技术的技术原理，并非旨在限制本专利技术的保护范围。
[0028]本专利技术公开了一种基于双子网络的红外弱小目标检测方法，如图1所示，包括如下步骤：
[0029]步骤1，分别构建子网络1和子网络2。子网络1设计思想包含两点：(1)为了避免图
[0048]网络2的损失函数采用了SmoothL1Loss，由于该网络设计旨在降低目标的虚警率，所以基于SmoothL1Loss，网络2的损失函数如下式所示：
[0049]MD2＝SmoothL1Loss(S2，G)
·
G
[0050]FA2＝SmoothL1Loss(S2，G)
·
(1
‑
G)
[0051]Loss2＝λ2MD2+FA2[0052]上述公式中：MD1和FA1分别代表网络1的漏检率和虚警率，S
i
和G分别代表网络1的分割结果和输入图像的真值(ground truth)，λ1是MD1的权重参数。
[0053]步骤4，分别训练子网络1和子网络2。采用8GB RAM，1.8GHz intel i7CPU处理器，Tesla P40 GPU的硬件环境；Ubuntu18.04，Python和Pytorch框架的软件环境。实验参数设置为：λ1＝100，λ2＝10；学习率每10次迭代降低20％，一共训练30个周期。
本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于双子网络的红外弱小目标检测方法，其特征在于，该方法包括以下步骤：步骤1，分别构建子网络1和子网络2的主体框架；步骤2，输入一批训练图像，经过数据预处理，送入到网络中；步骤3，针对子网络1，计算漏检抑制损失函数，更新子网络1参数；针对子网络2，计算虚警抑制损失函数，更新子网络2参数；步骤4，转至步骤2，直至训练收敛，得到最终的子网络1和子网络2；步骤5，在测试阶段，把测试图像分别输入最终的子网络1和子网络2，得到两组检测结果；步骤6，将来自于子网络1和子网络2的结果进行融合，输出最终检测结果。2.根据权利要求1所述的一种基于双子网络的红外弱小目标检测方法，其特征在于，所述步骤1中，将整个红外弱小目标检测任务分成两个子任务，即采用两个子网络来完成红外弱小目标检测任务；子网络1旨在降低检测的漏检率，即学习较低的阈值，使得网络的召回率升高；子网络2旨在降低检测的虚警率，即学习较高的阈值，导致网络的精确率升高，最后将两个网络输出的结果融合得到分割结果。3.根据权利要求1所述的一种基于双子网络的红外弱小目标检测...

【专利技术属性】
技术研发人员：杨万扣，王强，李红，王欢，
申请(专利权)人：东南大学，
类型：发明
国别省市：