基于RT-DETR的轻量化水下目标检测算法

Lightweight underwater object detection algorithm based on RT-DETR

  • 摘要: 水下目标检测是海洋探索与开发的关键技术,但水下环境的复杂性以及水下设备有限的计算资源给水下目标检测带来了巨大挑战。为此,本文提出一种轻量化的水下目标检测算法LSOR-DETR(Lightweight Spatial Online Reparameterization DETR)。首先,结合跨阶段局部网络的高效特征融合机制与加权卷积的自适应空间建模能力提出LSPA-Net(Lightweight Spatial Perception Aggregation Network),在保持主干网络特征表达能力的同时降低模型复杂度;其次,提出集成在线重参数化技术和RepNCSPELAN架构的ORFusion(OREPA-enhanced Restructured Fusion)模块,构建高效的特征融合机制;最后,引入GSConvE轻量下采样模块,弥补主干网络和颈部融合在轻量化过程中的精度损失。在DUO水下目标检测数据集上的实验结果表明,所提模型在保持RT-DETR模型的检测精度的同时,参数量和计算量分别降低了38.7%和40.7%。

     

    Abstract: Underwater object detection is a key technology for ocean exploration and development. However, the complexity of the underwater environment and the limited computing resources of underwater equipment pose enormous challenges to underwater object detection. To address these issues, this paper proposes a lightweight underwater object detection algorithm named LSOR-DETR (Lightweight Spatial Online Reparameterization DETR). Firstly, LSPA-Net (Lightweight Spatial Perception Aggregation Network) is proposed by combining the efficient feature fusion mechanism of Cross-Stage Partial Networks with the adaptive spatial modeling capability of weighted convolution, which reduces model complexity while preserving the feature representation ability of the backbone network. Secondly, the ORFusion (OREPA-enhanced Restructured Fusion) module integrating online reparameterization technology and the RepNCSPELAN architecture is presented to construct an efficient feature fusion mechanism. Finally, the GSConvE lightweight downsampling module is introduced to compensate for the precision loss caused by the lightweight process of the backbone network and neck fusion. Experimental results on the DUO underwater object detection dataset demonstrate that the proposed model maintains the detection accuracy of the RT-DETR model while reducing the number of parameters and computational cost by 38.7% and 40.7%, respectively.

     

/

返回文章
返回