采用改进Faster R-CNN的遥感图像目标检测方法

Remote sensing image target detection method using improved Faster R-CNN

  • 摘要: 遥感图像目标检测能为军事和民用领域提供重要的可利用信息,成为近年来的研究热点。针对现有目标检测技术不能兼顾检测速度和精度的问题,本文对Faster R-CNN做了优化:将轻量化的深度可分离残差网络作为Faster R-CNN的基础网络,降低基础网络模型的参数数量;将基础网络中的多层卷积特征经局部响应归一化后进行融合,增强目标特征信息的完备性,改善小目标易漏检的问题;联合softmax损失函数和中心损失函数训练网络模型,增加类别之间的差异性,缩小类内变化,使网络模型能学习到更具差异性的目标特征。在VEDAI、NWPU VHR-10、DOTA三个数据集上对本文方法进行验证,与传统Faster R-CNN相比,本文方法的检测精度提高了约7.0%。

     

    Abstract: Remote sensing image target detection can provide important usable information for military and civil fields and has become a research hotspot in recent years. Aiming at the problem that the existing target detection technology cannot give attention to both detection speed and accuracy, this paper optimized Faster R-CNN: the lightweight depth separable residual network was used as the backnone network of Faster R-CNN to reduce the number of parameters in the backnone network model. The multi-layer convolution features in the backnone network were fused after local response normalization to enhance the completeness of target feature information and improve the problem that small targets are easy to miss detection. The network model was trained by combining softmax loss function and center loss function to reduce the changes within categories and increase the differences between categories, so that the network model can learn more different target features. The proposed method was verified on VEDAI, NWPU VHR-10, DOTA datasets. Compared with the traditional Faster R-CNN, the detection accuracy of the proposed method is improved by 7.0%.

     

/

返回文章
返回