###
计算机系统应用英文版:2025,34(10):76-85
本文二维码信息
码上扫一扫!
改进YOLO11n的水下目标检测
(1.中北大学 软件学院, 太原 030051;2.山西云时代政务云技术有限公司, 太原 030032)
Improved Underwater Target Detection with YOLO11n
(1.School of Software, North University of China, Taiyuan 030051, China;2.Shanxi Cloud Era Technology Company Ltd., Taiyuan 030032, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 682次   下载 1420
Received:March 03, 2025    Revised:May 07, 2025
中文摘要: 针对水下环境光照不足、噪声干扰, 以及小目标的聚集导致的遮挡和误检、漏检问题, 提出了一种基于YOLO11n改进的水下目标检测模型. 提出小波池化卷积网络(WPCN), 通过小波池化技术分解特征, 保留低频子带, 避免频率混叠, 并且使用剪枝优化, 在保持推理速度的同时, 提高了检测精度. 引入单头视觉Transformer(SHSA)与卷积门控线性模块(CGCM), 提升对复杂场景的适应性. 提出共享可重参数化卷积检测头(RLD-Head), 通过共享卷积层减少参数量, 并利用重参数化技术避免性能损失, 适应资源受限设备. 设计了Wise-inner-MPDIoU损失函数, 提高了检测精度. 与原模型相比, mAP50提升了3.8个百分点, mAP50-95提升了4.3个百分点, 参数量减少了30.6%, 计算量减少了30.1%, 证明了该方法在水下目标检测方面的优势.
Abstract:To solve the problem of occlusion and false and missed detection caused by insufficient illumination and noise interference in the underwater environment, as well as the aggregation of small targets, this study proposes an improved underwater target detection model based on YOLO11n. Meanwhile, the wavelet pooling convolutional network (WPCN) is put forward to decompose features by wavelet pooling technology and preserve low-frequency subbands, thus avoiding frequency aliasing. Pruning optimization is adopted to improve detection accuracy while maintaining inference speed. Additionally, single head visual Transformer (SHSA) and convolutionally gated linear module (CGCM) are introduced to improve the adaptability to complex scenarios. A shared reparameterizable convolutional detection head (RLD-Head) is proposed to reduce the amount of parameters by sharing the convolutional layers, and reparameterization techniques are utilized to avoid performance loss and adapt to resource-constrained devices. The Wise-inner-MPDIoU loss function is designed to improve the detection accuracy. Compared with the original model, mAP50 improves by 3.8 percentage points, mAP50-95 increases by 4.3 percentage points, and the parameter amount and computation amount are reduced by 30.6% and 30.1% respectively, which proves the advantages of this innovation in underwater target detection.
文章编号:     中图分类号:    文献标志码:
基金项目:山西省自然科学基金面上项目(20210302124192)
引用文本:
王轩楷,潘广贞,陈新.改进YOLO11n的水下目标检测.计算机系统应用,2025,34(10):76-85
WANG Xuan-Kai,PAN Guang-Zhen,CHEN Xin.Improved Underwater Target Detection with YOLO11n.COMPUTER SYSTEMS APPLICATIONS,2025,34(10):76-85