[发明专利]一种基于改进YOLOv5的SAR图像飞机目标检测方法

说明书

本发明公开了一种基于改进YOLOv5的SAR图像飞机目标检测方法，涉及图像目标检测领域。收集获取包含飞机目标的SAR图像数据集，进行图像数据标注，划分为训练集和测试集；改进YOLOv5模型，搭建新的检测模型，包括输入层、主干网络、Neck网络、Head网络；设置好训练参数对检测模型进行训练，进行性能测试，采取精确度、召回率等评价指标来评估模型目标检测效果。本发明可以改进传统SAR飞机目标检测算法复杂背景下易漏检、虚检飞机的瓶颈，生成的飞机目标检测算法结合使用了引进注意力机制加强关注目标位置信息方法、特征融合方法以及全局信息关注方法，通过改进YOLOv5检测模型的主干网络、颈部结构、输出检测头来进行飞机目标的特征融合，提升飞机目标检测精度。

技术领域

本发明涉及图像目标检测领域，尤其涉及一种基于改进YOLOv5的SAR图像飞机目标检测方法。

背景技术

合成孔径雷达(SyntheticAperture Radar，SAR)是一种主动式微波成像传感器，具有全天时、全天候的成像观测能力。SAR自20世纪50年代诞生之后，由于其独特的成像机制，其在军事和民用领域都得到了广泛的应用。在民用领域，SAR在矿藏资源探测、灾情探测与防治、地形探测与绘制及农业、林业等方面发挥着巨大的作用；在军事领域，SAR图像在战场态势感知、典型目标识别和精准指导、隐形目标散射特性的静动态测量等方面具有突出的意义。由于SAR的独特性能，SAR图像目标解译技术受到了各国的重视，飞机作为其中一类重要的目标，在民用领域，飞机检测有助于机场的有效管理；在军事领域，获取飞机的数量、分布等信息具有重要价值。因此，利用SAR图像进行飞机目标的精确检测识别是个研究的热点方向并且具有重要意义。

近年来，随着深度神经网络的发展，目标检测模型分为one-stage检测模型和two-stage检测模型。双阶段检测模型先在图像中生产大量的候选区域，再通过卷积神经网络对这些区域进行特征提取，最后识别目标类别和位置。常用的双阶段目标检测模型有R-CNN、Fast R-CNN、Faster R-CNN等。基于候选区域的双阶段目标检测算法虽然精度很高，但是检测效率不高，而单阶段目标检测方法不需要生成候选区域，直接从输入图像中提取特征，继而预测物体的类别与位置信息。常用的单阶段目标检测模型有SSD和YOLO。

单阶段检测模型在速度上远远优于双阶段检测模型。在单阶段检测模型检测速度较快的情况下，检测时精度都没有达到预想的要求，在复杂背景中容易虚检、漏检目标。因此，通过改进检测模型进一步提升SAR图像飞机目标检测精度具有重要意义。

因此，本领域的技术人员致力于开发一种基于改进YOLOv5的SAR图像飞机目标检测方法，能够尽量避免漏检、误检，使得复杂背景下的飞机目标能够被精准检测。

发明内容

有鉴于现有技术的上述缺陷，本发明所要解决的技术问题是避免漏检、误检，精准检测复杂背景下的飞机目标。

为实现上述目的，本发明提供了一种基于改进YOLOv5的SAR图像飞机目标检测方法，对目标检测算法YOLOv5进行改进，获得改进后的目标检测算法YOLOv5，包括以下步骤：

步骤1、在输入层中采用Mosaic数据增强方式，对输入图像进行预处理，采用K均值聚类算法针对飞机目标尺寸优化锚框；

步骤2、所述步骤1的特征图输入改进的主干网络，改进的主干网络是指在主干网络中引入坐标注意力机制CA，采用SPD-CA卷积块替换主干网络对图像进行下采样所采用的跨步卷积；

步骤3、所述步骤2得到的不同尺度的图像特征图输入到改进的YOLOv5网络中的Neck部分，采用设计的FFP特征融合金字塔网络结构改进目标检测算法YOLOv5的Neck结构，实现自适应特征融合；

步骤4、所述步骤3得到不同尺度的张量数据，输入到改进的YOLOv5网络中的预测层部分，在预测层之前添加全局注意力机制GAM，捕捉全局信息，减少不准确的目标定位。