一种基于多尺度特征增强的单帧红外弱小目标检测方法

本发明公开了一种基于多尺度特征增强的单帧红外弱小目标检测方法，包括如下步骤：步骤1、获取单帧红外图像构建单帧红外图像数据集，并划分为训练集和测试集；步骤2、构建一个红外弱小目标检测网络模型，所述红外弱小目标检测网络以YOLOv5网络模型为基础网络，所述YOLOv5网络模型包括主干特征提取网络、特征融合网络和检测头；在所述主干特征提取网络中引入了多尺度特征增强模块，所述多尺度特征增强模块包括依次串联的上下文特征融合模块与特征细节增强模块；步骤3、设计训练策略，根据训练策略将训练集作为输入训练红外弱小目标检测网络模型；步骤4、完成训练后，输入测试集进行红外弱小目标检测网络模型的测试和评估。