计算机工程与应用2026,Vol.62Issue(1):101-111,11.DOI:10.3778/j.issn.1002-8331.2505-0249
改进YOLOv12的水面垃圾检测方法
Improving Water Debris Detection Method of YOLOv12
摘要
Abstract
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.关键词
水面垃圾检测/YOLOv12/WTConv/DynamicConv/WiseIoUv3Key words
water debris detection/YOLOv12/WTConv/DynamicConv/WiseIoUv3分类
信息技术与安全科学引用本文复制引用
朱宇凡,杨吉凯,李威..改进YOLOv12的水面垃圾检测方法[J].计算机工程与应用,2026,62(1):101-111,11.基金项目
国家自然科学基金(52171336). (52171336)