基于双支网络协作的红外弱小目标检测  

An Infrared Small Target Detection Method via Dual Network Collaboration

作　　者：王强[1,2,3] 吴乐天 李红[1,2] 王勇 王欢 杨万扣[1,2] WANG Qiang;WU Letian;LI Hong;WANG Yong;WANG Huan;YANG Wankou(School of Automation,Southeast University,Nanjing 210096,Jiangsu,China;Key Laboratory of Measurement and Control of Complex Systems of Engineering,Ministry of Education,Southeast University,Nanjing 210096,Jiangsu,China;Jiangsu Automation Research Institute,Lianyungang 222061,Jiangsu,China;School of Computer Science and Engineering,Nanjing University of Science and Technology,Nanjing 210094,Jiangsu,China)

机构地区：[1]东南大学自动化学院,江苏南京210096 [2]东南大学复杂工程系统测量与控制教育部重点实验室,江苏南京210096 [3]江苏自动化研究所,江苏连云港222061 [4]南京理工大学计算机科学与工程学院,江苏南京210094

出　　处：《兵工学报》2023年第10期3165-3176,共12页Acta Armamentarii

基　　金：国家自然科学基金项目(62276061)。

摘　　要：红外弱小目标检测在预警系统和导弹制导中具有重要的作用,一直是红外图像处理中颇受关注的研究方向。由于红外弱小目标具有信杂比低、尺寸小、形状结构不明显和纹理弱等特点,现有的通用目标检测和语义分割网络直接应用到红外弱小目标检测效果不佳,为此提出一种基于双支网络协作的红外弱小目标检测网络(DualNet)。将检测任务划分成两个子任务,即降低漏检和降低虚警,进而设计两个不同的网络架构分别处理,并利用加权融合损失函数将两支网络信息整合,使得DualNet能够有效地平衡漏检率和虚警率。在自建数据集上的实验结果表明:DualNet相较于通用性能较好的FCN、DeepLabv3、cGAN以及U-net语义分割网络模型具备更高的准确率和鲁棒性,其在F1-measure指标上提高了8%;在SIRST公开数据集上的检测性能也显著超过了基于深度学习的红外目标检测模型ACM和MDvsFA-cGAN,以及多个经典的非深度学习红外弱小目标检测方法。研究结果表明,所提出的方法能够有效提高红外弱小目标的检测精度。Infrared small target detection(ISTD)is a heated topic in infrared image processing,and it is intensively applied in early warning systems and missile guidance.ISTD faces significant challenges such as low signal-to-noise ratio(SNR),small size,lack of distinct shape or structure,and weak texture,making it a demanding task.The performance of conventional object detection networks and semantic segmentation networks considerably deteriorates when applied directly to ISTD tasks.To address this issue,this paper proposes a new dual network collaboration-based image semantic segmentation network for ISTD,termed as DualNet.DualNet divides the task into two sub-tasks,namely reducing missed detections and reducing false alarms,with two sub-networks focusing on their respective targets(with cost reduced)by employing a weighted loss function to integrate sub-network information.DualNet effectively balances the miss detection rate and false alarm rate.Experimental results show that DualNet outperforms general neural network models(e.g.FCN,DeepLabv3,cGAN and U-net)on the ISTD task,with an improved F1-measure by 0.08.Furthermore,our model outperforms ACM and MDvsFA-cGAN,two most representative ISTD models based on deep learning,and several non-deep-learning-based ISTD methods.

关 键 词：红外弱小目标 目标检测 双支网络协作 语义分割 深度学习 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]