计算机与现代化 ›› 2021, Vol. 0 ›› Issue (02): 35-39.

• 图像处理 • 上一篇    下一篇

FRDet:一种基于候选框特征修正的多方向遥感目标快速检测方法

  

  1. (华北计算技术研究所基础四部,北京100083)
  • 出版日期:2021-03-01 发布日期:2021-03-01
  • 作者简介:涂鑫(1996—),男,江西吉安人,硕士研究生,研究方向:目标检测,E-mail: 2197654067@qq.com; 王滨(1977—),男,山东烟台人,硕士,研究方向:目标检测,无人系统。

FRDet: A Fast Detection Method for Multi-directional Remote Sensing Targets Based on Feature Correction of Candidate Frames

  1. (Department 4 of Foundation, North China Institute of Computing Technology, Beijing 100083, China)
  • Online:2021-03-01 Published:2021-03-01

摘要: 在遥感图像的目标检测任务中,为了能更加准确地定位目标,现有的基于候选框特征提取的one-stage检测方法是在每个空间位置上,充分预设多个候选框从而覆盖住待检测目标,然而这会造成one-stage检测方法计算复杂度的大幅提升。本文提出一种基于候选框特征修正的多方向遥感目标检测方法。在该方法中,特征图的每个位置仅预设一个候选框,通过回归学习进行特征修正后得到的候选框替换掉原始的框,再由one-stage检测方法的分类层和回归层分别进行识别和定位。所提方法采用Mobilenetv2作为检测网络的基本结构,在DOTA数据集上飞机的检测率可达96.8%,虚警率为6.7%,mAP值达0.87,并且具有完全的实时结果,超过了SSD、YOLOv3等所有基于候选框特征提取的的遥感图像检测方法。由于本文方法巧妙地避开了候选框的宽高比和尺度的先验设计,因此本文方法很容易应用于其他类似的检测任务中,即插即用,具有很强的任务适应性。

关键词: 遥感图像, 目标检测, 特征修正

Abstract: In the target detection task of remote sensing images, in order to locate the target more accurately, the existing one-stage detection method based on the feature extraction of candidate frames is to fully preset multiple priori frames at each spatial position to cover the target to be detected. However, this will greatly increase the computational complexity of the one-stage detection method. This paper proposes a multi-directional remote sensing target detection method based on the feature correction of the candidate frame. In this method, only one candidate frame is preset at each position of the feature map. We replace the original box with the candidate box obtained after feature correction through regression learning, and then use the classification layer and regression layer of the one-stage detection method to identify and locate respectively. The method uses Mobilenetv2 as the basic structure of the detection network. The detection rate of aircraft on the DOTA dataset can reach 96.8%, the false alarm rate is 6.7%, the mAP value is 0.87, and it has complete real-time results, surpassing all remote sensing image detection methods based on candidate frame feature extraction such as SSD and YOLOv3. Because this method cleverly avoids the priori design of the aspect ratio and scale of the candidate frame, this method can be easily applied to other similar detection tasks, plug-and-play, it has strong task adaptability.

Key words: remote sensing image, target detection, feature correction