改进YOLOv4的遥感图像目标检测算法

Improved YOLOv4 algorithm for remote sensing image object detection

为有效解决遥感图像目标检测算法在复杂背景下的检测效果不佳的问题,提出一种改进YOLOv4的目标检测算法。设计一种跨阶段残差结构,替换原主干网络的简单残差结构,降低模型参数量和计算负担;引入CBAM注意力机制,加强CSP模块间有效特征交互;使用跨阶段分层卷积模块重构特征融合阶段对深层特征图的处理方式,防止网络退化和梯度消失;采用Mish激活函数,增强融合网络对非线性特征的提取能力。在RSOD、DIOR数据集上的实验结果表明,改进YOLOv4算法的测试mAP相比原YOLOv4算法分别高出4.5%、7.3%,其检测速度分别达到48 fps、45 fps,在保证实时性的同时检测精度有较大提升。

To effectively solve the problem of poor detection effects of remote sensing image target detection algorithm in complex background,an improved YOLOv4 target detection algorithm was proposed.A cross-stage residual structure was designed to replace the simple residual structure in the CSP module of the original backbone network to reduce the amount of model parameters and computational burden.The CBAM attention mechanism was introduced to strengthen the interaction of effective features between CSP modules.The cross-stage hierarchical convolution module was used to reconstruct the processing method of the deep feature map in the feature fusion stage to prevent network degradation and gradient disappearance.The Mish activation function was used to enhance the extraction ability of the fusion network for nonlinear features.Experimental results on the RSOD and DIOR data sets show that the test mAP of the improved YOLOv4 algorithm is 4.5%and 7.3%higher than that of the original YOLOv4 algorithm,and its detection speed reaches 48 fps and 45 fps respectively,which ensure the real-time performance and greatly improve the detection accuracy.