基于改进RT-DETR的浅水海洋生物识别方法

来源：电子测量技术杂志2024年第18期北京时间：

作者：蒋智臣，胡俐蕊

单位：1.广西大学计算机与电子信息学院 南宁 530004; 2.北部湾大学电子与信息工程学院 钦州 535011

摘要：针对现有浅水海洋生物识别方法在水下环境中对浅水海洋生物识别效果不佳的问题，提出了一种以RT-DETR为基准模型的改进浅水海洋生物识别方法。首先，使用重参数化网络RepViT作为模型的主干网络，提升模型的特征提取能力。然后，构建基于重参数化的并行膨胀卷积RepPDC并引入颈部网络中，使模型能够有效获取长距离上下文信息，有利于提升模型的识别精度。最后，基于注意力机制构建了双向特征融合模块CAFM，提升模型在水下环境中对重点信息的关注能力。实验结果表明，改进后的方法，mAP50提升至87.5%，mAP75提升至70.9%，mAP50:95提升至64.9%，且参数量更少，有望应用到实际浅水海洋生物识别任务中。

关键词：海洋生物识别;目标检测;深度学习;RT-DETR

基金资助：广西区科技计划项目(桂科AC17195057)、广西钦州市科技计划项目(202116602)资助