• • 上一篇
邓健志1, 唐政豪2, 3, 李云1
2. 桂林理工大学, 光电信息与智能通信技术工程研究中心, 广西 桂林 541006
3. 桂林理工大学, 广西高校低维结构物理与应用重点实验室, 广西 桂林 541006
DENG Jianzhi1, TANG Zhenghao2, 3, LI Yun1
2. Guilin University of Technology, College of Physics and Electronic Information Engineering, Guilin 541006, China; Guilin 541006, China;
3.Guilin University of Technology, Key Laboratory of Low-dimensional Structural Physics and Application, Education Department of Guangxi Zhuang Autonomous Region, Guilin 541006, China
摘要: 浅海底栖生物目标检测在海洋生态监测和资源管理中具有重要意义,但受限于水下图像的低光照、模糊及复杂背景,传统检测算法性能不佳。本文提出一种改进的RT-DETR模型(MEIE-RTDETR),设计了一种多尺度边缘信息增强(multiscale edge information enhance, MEIE)模块强化特征提取,采用自适应稀疏自注意力(adaptive sparse self-attention, ASSA)降低计算冗余,并引入亮度信息增强(intensity enhance layer, IELC3)模块改进特征金字塔提升小目标检测能力,最后设计P-IoU(Powerful-IoU)+NWD(Normalized Wasserstein Distance)损失函数增强对边界框模糊和多尺度目标检测的性能。在DUO和RUOD数据集上的实验表明,改进后的模型在参数量(15.83M)和计算量(49.5 GFLOPs)显著降低的同时,mAP@50分别达到85.0%和85.4%,优于Faster R-CNN、YOLO系列及原始RT-DETR,为水下轻量化高精度检测提供了有效解决方案。