本文已被:浏览 574次 下载 1222次
Received:March 13, 2025 Revised:May 07, 2025
Received:March 13, 2025 Revised:May 07, 2025
中文摘要: 针对水下检测背景复杂、光线暗淡、目标遮挡重叠等问题, 本文提出一种基于YOLOv8n改进的实时水下目标检测算法. 首先, 构造特征融合模块SEHAP, 使P2层与P3层融合, 再经过EHAPOKM学习全局到局部的特征, 以提高识别小目标和低分辨率图像的准确率. 其次, 增加一个轻量化的检测头SLDH, 使用共享卷积, 并将其与ASL缩放特征尺度模块结合, 在降低参数量和计算量的情况下, 减少精度损失. 随后, 增加C2f-EGMSC模块, 采用分组卷积更好地提取不同层次的特征. 最后在部分C2f-EGMSC模块后使用注意力机制BAM, 使模型同时关注通道和空间维度信息, 提升模型性能, 将改进后的模型命名为ESE-YOLOv8. 基于RUOD数据集上的实验结果表明ESE-YOLOv8可以达到85.2%的检测精度, 相较于原始算法提升1.2个百分点, 参数量下降了36.7%. 改进后的模型兼顾了轻量化和精度, 为水下环境部署提供了可行的解决方案.
Abstract:To address the challenges of underwater target detection, including complex backgrounds, low illumination, and frequent occlusion and overlap of targets, this study proposes an improved real-time underwater target detection algorithm based on YOLOv8n. A feature fusion module, SEHAP, is first designed to fuse layers P2 and P3. This fused output is then processed by EHAPOKM to learn features ranging from global to local, thus improving the detection accuracy for small targets and low-resolution images. Moreover, a lightweight detection head, SLDH, is introduced. By employing shared convolution and integrating it with the ASL feature scaling module, the proposed method reduces both parameter count and computational load with minimal accuracy loss. The C2f-EGMSC module is further incorporated, utilizing group convolution to enhance multi-level feature extraction. To improve the model’s capacity to capture spatial and channel-wise information, the bottleneck attention module (BAM) is applied after selected C2f-EGMSC modules. The improved model, named ESE-YOLOv8, is evaluated on the RUOD dataset. Experimental results demonstrate a detection accuracy of 85.2%, representing a 1.2 percentage point improvement over the baseline, with a 36.7% reduction in parameters. The improved model takes into account both lightweight and precision, providing a feasible solution for deployment in underwater environments.
keywords: target detection YOLOv8n shared convolution lightweight multi-scale feature fusion attention mechanism
文章编号: 中图分类号: 文献标志码:
基金项目:国家重点研发计划(2020YFA0608000)
引用文本:
肖潇,方睿,朱洪廷,蔡鑫地.基于改进YOLOv8n的轻量化水下目标检测.计算机系统应用,2025,34(10):184-194
XIAO Xiao,FANG Rui,ZHU Hong-Ting,CAI Xin-Di.Lightweight Underwater Target Detection Based on Improved YOLOv8n.COMPUTER SYSTEMS APPLICATIONS,2025,34(10):184-194
肖潇,方睿,朱洪廷,蔡鑫地.基于改进YOLOv8n的轻量化水下目标检测.计算机系统应用,2025,34(10):184-194
XIAO Xiao,FANG Rui,ZHU Hong-Ting,CAI Xin-Di.Lightweight Underwater Target Detection Based on Improved YOLOv8n.COMPUTER SYSTEMS APPLICATIONS,2025,34(10):184-194

