基于改进SSD的遥感目标检测算法

REMOTE SENSING TARGET DETECTION ALGORITHM BASED ON IMPROVED SSD

  • 摘要: 在遥感图像目标检测领域,目标特征的提取以及多尺度特征融合是研究难点,为解决上述难点,该文提出一种优化SSD网络的遥感图像目标检测算法。该算法在SSD网络中引入改进RFB模块,提升算法的特征提取能力;设计多尺度注意力特征融合模块,增强对多尺度特征信息的表达能力;利用改进K-means聚类算法重新设计锚框匹配策略,加强网络对特征的读取能力。实验结果表明,该文算法的平均检测精度mAP达到95.16%,相对于原始SSD算法提高16.36百分点,且与其他改进算法相比,精度明显提高,验证了该算法的优越性。

     

    Abstract: In the field of remote sensing image target detection, the extraction of target features and multi-scale feature fusion are the research difficulties. In order to solve the above difficulties, this paper proposes a remote sensing image object detection algorithm based on optimized SSD network. An improved RFB module was introduced into the SSD network to improve the feature extraction capability of the algorithm. A multi-scale attention feature fusion module was designed to enhance the expression ability of multi-scale feature information. The anchor box matching strategy was redesigned by using the improved K-means clustering algorithm to strengthen the network’s ability to read features. Experimental results show that the average detection accuracy mAP of the proposed algorithm reaches 95.16%, which is 16.36% higher than the original SSD algorithm, and the accuracy is significantly improved compared with other improved algorithms, which verifies the superiority of the proposed algorithm.

     

/

返回文章
返回