• 多尺度特征增强融合的ME-RTDETR水下目标检测

    Multi-scale feature enhancement and fusion of ME-RTDETR for underwater object detection

    • 水下目标检测常因水下图像噪声较大且包含多尺度目标混合等问题而导致漏检、错检现象。为此,提出了一种基于RTDETR(Deal - Time Detection Transformer)改进的多尺度特征增强融合的ME-RTDETR(Multi-scale feature Enhancement fusion RTDETR)水下目标检测算法。首先,使用CSPnet替代原始主干网络,提升模型的特征提取能力。其次,提出SHViTCGLU模块替换C2f中的残差模块,通过增强局部特征与全局特征提取,提升模型处理噪声图像的能力。再次,基于原始CCFM模块设计多尺度特征增强融合金字塔结构,增强后的特征层和细粒度的特征融合使算法模型能更准确地定位和识别不同尺度的目标。最后,设计Focaler-MPDIoU损失函数替换原模型的GIoU损失函数,能更好地学习边界框的位置和形状,加速模型收敛。实验结果表明:改进后的模型相较于基线模型在DUO数据集上准确率提升了4.5%,平均精度值mAP@0.5提高了2.1%,参数量降低了26%。所提算法在提高平均检测精度、降低模型参数量均有明显改进,验证了其在水下目标检测任务中的实用价值。

       

      Abstract: In the process of underwater target detection, the presence of significant noise in underwater images and the mixture of multi-scale targets can easily lead to missed detections and false positives. An improved underwater target detection algorithm called ME-RTDETR(Multi-scale feature Enhancement fusion RTDETR)is proposed, which is based on RTDETR(Deal-Time Detection Transformer) with enhanced multi-scale feature fusion. Firstly, CSPnet is used to replace the original backbone network to enhance the model's feature extraction capabilities. Secondly, the SHViTCGLU module is introduced to replace the residual module in C2f. This module enhances both local and global feature extraction, thereby improving the model's ability to handle noisy images. Thirdly, a multi-scale feature enhancement fusion pyramid structure is designed based on the original CCFM module. The enhanced feature layers and fine-grained feature fusion enable the model to more accurately locate and recognize targets of different scales. Finally, the Focaler-MPDIoU loss function is designed to replace the original GIoU loss function, which allows the model to better learn the position and shape of bounding boxes and accelerates convergence. Experimental results show that the improved model has an accuracy increase of 4.5% compared to the baseline model on the DUO dataset, an average precision value mAP@0.5 increase of 2.1%, and a parameter reduction of 26%. The proposed algorithm has significant improvements in improving average detection accuracy and reducing model parameters, verifying its practical value in underwater target detection tasks.

       

    /

    返回文章
    返回