稠密连接递归特征金字塔的遥感目标检测算法

Object detection in remote sensing images using densely connected recursive feature pyramids

针对遥感目标检测中,目标尺寸较小、相似地物易混淆和背景复杂干扰大等问题,提出了一种基于稠密连接递归特征金字塔的遥感目标检测算法。首先,为充分利用遥感图像特征对特征融合模式进行了改进,利用典型相关分析CCA(Canonical Correlation Analysis),取代较为简单的逐像素相加融合模式,增强特征融合有效性;其次,为加强对小尺度目标的特征提取加入多感受野机制,利用不同尺寸的空洞卷积提取并融合不同感受野的特征,增强网络感知力;接着,为解决遥感多尺度目标泛化问题对特征递归形式进行改进,引入稠密连接结构,增强特征融合密度,充分利用骨干网络与高低层特征信息;最后,在前文基础上构建稠密连接递归特征金字塔模型DR-FPN(Densely-connected Recursive Feature Pyramid Network),并利用融合信息实现对遥感目标的精准定位。实验结果表明,在通用数据集MS-COCO2017上,使用本文金字塔模型平均精度可以提升9.9%;在遥感数据集NWPU VHR上,使用本文算法平均精度可以提升1.1%;在遥感数据集DIOR上,使用本文算法平均精度可以提升2.2%,超过其他特征金字塔模型和检测算法;在大规模遥感数据集DOTA上,使用本文算法平均精度可以提升1.8%,超过其他特征金字塔模型和检测算法,实现了对遥感目标的高精度检测。

In recent years,the multiscale utilization of input sample features has gradually become a research hotspot in the field of target detection.However,remote sensing target detection suffers from some problems,such as small target size,easy confusion with similar objects,and extensive background interference.Therefore,a remote sensing target detection algorithm based on dense connection recursive feature pyramids is proposed.First,the feature fusion mode is improved to use the features of remote sensing images fully.The traditional feature fusion method is only pixel-by-pixel addition,which is simple and rough to calculate and cannot effectively screen features.Therefore,canonical correlation analysis is used to replace the simple pixel-by-pixel additive fusion mode to enhance the effectiveness of feature fusion.Moreover,this method does not add any new parameters.Second,the multireceptive field(MRF)mechanism was added to enhance the feature extraction of small-scale targets,and the features of different receptive fields were extracted and fused by dilated convolution of different sizes to enhance network perception.Given the increase in receptive field types,the richness of features that can be extracted is greatly enhanced,which is conducive to the improved transmission of effective information.In addition,our proposed MRF module is a multibranch convolution module,which is intended to mimic the human visual receptive field mechanism.Then,the feature recurrence form is improved to solve the generalization problem of a multiscale remote sensing target,and the dense connection structure is introduced to enhance the feature fusion density.A dense connection improves network performance because the feature level increases,and the feature richness is enhanced accordingly.Compared with the original recursive feature pyramid,the utilization of the backbone network is remarkably improved.The backbone network and the feature information of high and low layers are fully utilized.Finally,Based on our proposed methods above,this study changes the way of feature recursion and designs a dense connection structure of the recursive feature pyramid.That is,it adds dense connections between multiscale features and each layer of the backbone network to improve the efficiency of feature extraction and utilization.In summary,the network design in this study includes a top-down fusion subnetwork,bottom-up path enhancement subnetwork,and feature recursive fusion subnetwork.Experimental results show that the average accuracy of the proposed pyramid model can be improved by 9.9%on the general dataset MS-COCO2017.On the remote sensing dataset NWPU VHR,the average accuracy of the proposed algorithm can be improved by 1.1%.On the remote sensing dataset DIOR,the average accuracy of the proposed algorithm can be increased by 2.2%,which is higher than other feature pyramid models and detection algorithms.On the large-scale remote sensing dataset DOTA,the average accuracy of the proposed algorithm can be increased by 1.8%.Experimental results show that the proposed method can outperform other feature pyramid models and detection algorithms.It achieves not only high precision detection of remote sensing targets but also has good performance on the benchmark dataset COCO.Therefore,the proposed method is advanced.