[发明专利]一种基于特征对齐的遥感场景下旋转目标检测方法

说明书

本发明公开了一种基于特征对齐的遥感场景下旋转目标检测方法，首先生成训练集，然后在Residual Network和Feature Pyramid Networks的基础上构建目标检测网络，对输入样本图像进行多尺度特征提取，再设计特征对齐模块，将由Feature Pyramid Networks网络得到的回归分支特征和分类分支特征进行特征融合，获得特征对齐后的有效特征，然后再经过由粗到细的检测框回归模块，所提取的特征利用通用卷积生成粗糙的检测框，之后再利用可变形卷积微调粗糙检测框，最终得到定位更加准确的检测框；最后对旋转检测网络进行训练，得到训练好的模型用于实际场景中的旋转目标检测任务。本发明相比现有技术直接预测检测框，检测框位置更加精确，算法的检测速度得到明显的提升。

技术领域

本发明属于图像处理技术领域，具体涉及一种旋转目标检测方法。

背景技术

遥感场景下的旋转目标检测是为了判断遥感图像中是否存在目标，并对其进行精确有向定位和分类，在军事和民用领域都具有重要应用。具体地，在军事方面，其可以用于国土入侵监控、军事侦察、军事打击等；在民用方面，可以用于城市交通监控、土地规划、灾情监测等诸多应用。与自然图像中的目标检测任务相比，遥感场景下的目标检测具有以下特点：(1)遥感图像分辨率更高、数据量庞大。(2)遥感目标普遍尺寸较小、密集且具有旋转性。其中，诸如舰船、飞机、汽车等旋转目标是遥感影像的典型检测目标。(3)由于传感器自身物理技术特性、观察视角和成像机理等原因，遥感平台获取的影像通常会混有噪声，并且易受天气、光照等多种因素影响。这些复杂特性使得遥感图像的目标检测充满挑战。因此，如何精确地对遥感场景下的旋转目标进行精确定位和分类具有重要的研究意义。目前，主流的旋转检测算法都是基于深度学习，按照检测阶段的不同，可以将相关工作分为以下两类：

第一种为两阶段的旋转目标检测算法。该类方法主要是基于区域框架进行检测。具体地，第一阶段从图像中生成与类别无关的带有方向性的区域建议，然后对这些区域进行特征提取；第二阶段使用与类别相关的分类器和回归器进行分类和回归；最后利用非极大抑制(Non-Maximum Suppression)等后处理方法得到检测结果。其中，Ding等人在论文“J.Ding,N.Xue,Y.Long,G.Xia,and Q.Lu,Learning roi transformer for detectingoriented objects in aerial.Proc.IEEE Conference on Computer Vision andPattern Recognition,pp.2849–2858,2019”中设计了Rotated Region of Interest(RRoI)将水平感兴趣区域(Horizontal Region of Interest)转换为旋转感兴趣区域(Rotated Region of Interest)。基于RRoI提出了Rotated Position Sensitive RoIAlign(RPS-RoI-Align)模块，在提取旋转不变特征同时实现特征对齐，以促进后续分类和回归。Yang等人在论文“X.Yang,J.Yang,J.Yan,Y.Zhang,T.Zhang,Z.Guo,X.Sun,and K.Fu,Scrdet:Towards more robust detection for small,cluttered and rotatedobjects.Proc.IEEE Conference on International Conference on Computer Vision,pp.8232–8241,2019”中设计了一种采样融合网络，它将多层特征融合到有效的锚点采样中，以提高对于小型目标的检测灵敏度。与此同时，通过抑制噪声和突出物体的特征，将有监督的像素注意力网络和通道注意力网络，用于小而杂乱的目标检测。Xu等人在论文“Y.Xu,M.Fu,Q.Wang,Y.Wang,K.Chen,G.Xia,and X.Bai,Gliding vertex on thehorizontal bounding box for multi-oriented object detection.IEEE Transactionson Pattern Analysis and Machine Intelligence,2020”中通过改变旋转框的表示方式避免了排序的问题。具体地，该方法先检测水平框，然后通过学习水平框四个角点的偏移量达到四边形检测的目的。由于这些方法需要先对建议区域进行提取，所以会极大地影响检测速度。