改进YOLOv12的水面垃圾检测方法OA
Improving Water Debris Detection Method of YOLOv12
针对现有水面垃圾检测方法模型较老、易漏检错检、无法满足高效的水面环境治理的问题,提出了改进YOLOv12的水面垃圾检测方法WDW-YOLO.该方法引入基于小波变换的WTConv改进颈部网络的C3k2模块,设计C3k2_WTConv模块,在保证模型参数可控性的同时实现了模型感受野的有效扩大;对于骨干网络中的C3k2模块,引入DynamicConv进行创新得到C3k2_DynamicConv模块,增加更多可学习参数的同时限制计算量的增长,增强模型对特征信息的捕捉能力,进而提高检测精度;将CIoU替换为WiseIoUv3并探究最佳的超参数组合,平衡检测难度不同的样本,增强模型的小目标检测能力.在FloW_IMG数据集上进行对比实验和消融实验,实验结果表明,提出的WDW-YOLO模型较基线YOLOv12n模型mAP50和mAP50-95分别提高2.7和1.8个百分点,达到90.7%和50.6%,在降低模型复杂度的同时实现了更高的检测准确率,在自动化的水面清理和水体保护中具有高可靠性与高实用性.
To address the issue of outdated models,high rates of missed and false detections,and the inability of existing water surface debris detection methods to meet the requirements of efficient water surface environmental management,this paper proposes an improved water debris detection method,WDW-YOLO,based on YOLOv12.This method intro-duces WTConv based on wavelet transform to improve the C3k2 module in the neck network,designing the C3k2_WTConv module.This design effectively expands the model's receptive field while maintaining controllable model parameters.For the C3k2 module in the backbone network,it introduces DynamicConv to innovate and obtain the C3k2_DynamicConv module.This approach increases the number of learnable parameters while limiting the growth of computational load,thereby enhancing the model's ability to capture feature information and improving detection accuracy.Additionally,it replaces CIoU with WiseIoUv3 and explore the optimal combination of hyperparameters to balance samples with different detection difficulties and enhance the model's ability to detect small objects.Experiments are conducted on the FloW_IMG dataset,including comparative and ablation studies.The results show that the proposed WDW-YOLO model achieves a 2.7 and 1.8 percentage points increase in mAP50 and mAP50-95,respectively,compared to the baseline YOLOv12n model,reaching 90.7%and 50.6%.This improvement is achieved while reducing model complexity and enhancing detec-tion accuracy.The WDW-YOLO model demonstrates high reliability and practicality for automated water surface cleaning and water body protection.
朱宇凡;杨吉凯;李威
华中科技大学 电气与电子工程学院,武汉 430074华中科技大学 船舶与海洋工程学院,武汉 430074华中科技大学 船舶与海洋工程学院,武汉 430074
信息技术与安全科学
水面垃圾检测YOLOv12WTConvDynamicConvWiseIoUv3
water debris detectionYOLOv12WTConvDynamicConvWiseIoUv3
《计算机工程与应用》 2026 (1)
101-111,11
国家自然科学基金(52171336).
评论