优化特征融合的多尺度遥感图像目标检测方法

MULTI-SCALE REMOTE SENSING IMAGE TARGET DETECTION METHOD BASED ON OPTIMIZED FEATURE FUSION

  • 摘要: 为解决遥感图像场景下多尺度目标检测准确率低的问题,提出DAFFNet遥感图像目标检测算法。该算法基于SSD进行了三方面的改进:为增强多尺度特征信息的获取能力,设计一种基于分组的特征融合方法;引入基于注意力机制的多维度特征优化方法,来解决复杂背景下目标分类困难的问题;将Focal loss作为新的边界框置信度损失函数,令模型聚焦于难分类的正样本,以改善正负样本不平衡对目标分类所造成的干扰。在遥感公共数据集NWPU VHR-10 上进行模型评估,实验结果表明,该算法相较于原算法均值平均精度提高5.1百分点,能有效地提高遥感图像目标检测准确率。

     

    Abstract: A DAFFNet remote sensing image object detection algorithm is proposed to solve the problem of low accuracy of multi-scale object detection in the remote sensing image scene. Based on SSD, the algorithm was improved in three aspects. We designed a group-based feature fusion method to enhance the ability to acquire multi-scale feature information. A multi-dimensional feature optimized method based on the attention mechanism was introduced to solve the difficulty of target classification in a complex background. The focal loss was used as a new bounding box confidential loss function to make the model focus on the positive samples that were difficult to classify, so as to improve the interference caused by the imbalance of positive and negative samples to target classification. The model was evaluated on the remote sensing public dataset NWPU VHR-10. The experimental result shows that the proposed algorithm improves the mean average precision by 5.1 percentage points compared with the original algorithm, which can effectively increase the object detection accuracy of remote sensing image.

     

/

返回文章
返回