首页|期刊导航|桂林电子科技大学学报|基于混合局部通道注意力机制的水下光学图像目标检测算法

基于混合局部通道注意力机制的水下光学图像目标检测算法OA

Underwater optical image target detection algorithm based on mixed local channel attention mechanism

中文摘要英文摘要

在水下光学图像目标检测任务中,水下环境的复杂性、光线的衰减以及水下生物多以小目标形态呈现共同影响了水下目标的检测精度.为了提高水下目标检测精度,提出一种基于YOLOv5s的水下光学图像目标检测算法.首先,在主干网络上引入Spd-Conv模块,该模块增强了对小目标的识别能力,从而有效提升了模型的检测精度;其次,在预测网络中加入YOLOx_head模块,通过使用解耦的检测头,提升了模型的收敛速度,使模型在训练时快速收敛;最后,在原有的混合局部通道注意力机制的基础上,设计了捕捉局部空间信息的C3-MLCA模块,从而增强了对目标特征的捕捉能力,并进一步提高了模型的检测精度.实验结果表明,该算法的mAP@0.5 提升了 2 个百分点,达到 84.1%;mAP@0.5:0.95 提升了 3 个百分点,达到 48.0%.

In underwater optical image target detection tasks,complex underwater environments,light attenuation,and the preva-lence of small underwater targets significantly affect detection accuracy.To improve this accuracy,an underwater target detection al-gorithm based on YOLOv5s is proposed.Firstly,the Spd-Conv module is introduced into the backbone network to enhance the recognition capability of small targets,effectively improving the detection accuracy.Secondly,the YOLOx_head module is added to the prediction network to enhance the model's convergence speed by using decoupled detection heads,enabling rapid convergence during training.Finally,building upon the existing Mixed Local Channel Attention mechanism,the C3-MLCA module is designed to capture local spatial information,thereby enhancing the model's ability to capture target features and further improving detection accuracy.Experimental results demonstrate that the improved algorithm increases a 2.0%increase mAP@0.5 by 2.0%,reaching 84.1%,and improves mAP@0.5:0.95 by 3.0%,reaching 48.0%.These results confirm the effectiveness of the proposed algorithm and the enhancement in detection accuracy.

陈辉;王奎阳;张兆贤

桂林电子科技大学 信息与通信学院,广西 桂林 541004桂林电子科技大学 信息与通信学院,广西 桂林 541004桂林电子科技大学 信息与通信学院,广西 桂林 541004

信息技术与安全科学

YOLOv5s目标检测光学图像注意力机制深度学习

YOLOv5starget detectionoptical imageattention mechanismdeep learning

《桂林电子科技大学学报》 2026 (1)

43-50,8

国家自然科学基金(62361018)广西自然科学基金(2021JJA170177)桂林电子科技大学研究生教育创新计划(2022YCXS027)

10.16725/j.1673-808X.202449

评论