联合特征增强和锚点自动生成的遥感图像高精度目标检测
High precision object detection in remote sensing images by combining feature enhancement and anchor automatic generation
-
摘要: 目标检测是遥感图像处理领域中一项重要而具有挑战性的任务,针对遥感图像中目标尺度差异较大以及方向分布随机等导致的遥感图像多尺度目标检测精度较低问题,本文提出一种基于特征增强和锚点框自动生成模块的目标检测方法。该方法在ResNet50网络中加入可操控的空洞卷积模块,并以此为基础设计了增强特征金字塔网络,提高网络对于目标多尺度特征表达能力。在区域建议网络中利用锚点框自动生成模块自主学习锚点框的位置和形状,以此获得更为稀疏和高质量的候选区域。本文在NWPU VHR-10数据集和飞机目标数据集上与多种基于卷积神经网络的目标检测算法进行对比实验,结果表明,本文所提方法在两个数据集上的mAP均为最优,分别为99.2%和87.7%,该方法具有较强的尺度自适应能力,有效的提高了遥感图像多尺度目标检测的精度。
Abstract: Object detection is an very important and challenging tasks in the field of remote sensing image processing. To solve the problem of low accuracy of multi-scale object detection in remote sensing image caused by differences of object scale and random directional distribution, this paper proposed a object detection method based on feature enhancement and anchor automatic generation module. In this method, a controllable atrous convolution module is added into the Resnet50 network, and based on this, an enhanced feature pyramid network is designed to improve the multi-scale feature representation ability of the network. In the region generation network, the anchor automatic generation module is used to learn the position and shape of the anchors independently, so as to obtain more sparse and high quality candidate regions. The comparison experiments and analyzes between proposed method and some other detection algorithm based on convolution neural network on the NWPU VHR-10 dataset and aircraft object dataset, the results show that the mAP of the proposed method is optimal on the two datasets, which are 99.2% and 87.7%, respectively. This method has strong scale adaptive ability and effectively improves the accuracy of multi-scale object detection in remote sensing images.