Di Juxing, Xi Kaikai, Yang Yang
Hebei University of Architecture, Information Engineering College, Zhangjiakou, 075000, China.
Sci Rep. 2025 Jul 11;15(1):25015. doi: 10.1038/s41598-025-10163-2.
To address the challenges in floating waste detection on water surfaces, such as small object scale, irregular shapes, and strong background interference, this study proposes an enhanced detection model based on the YOLOv8s frame work, named ES-YOLOv8. The new model optimizes the feature fusion strategy in the neck, constructing a refined "160-80-40-20" multiscale detection frame work. Integrated with the Efficient Multiscale Attention (EMA) module, it significantly improves the model's ability to extract features of small float ing objects. Additionally, an innovative Shape-IoU loss function is employed to optimize the bounding box regression accuracy of irregular targets through shape-sensitive constraints. This results in the development of an enhanced model that integrates feature enhancement, interference suppression, and localization optimization. Experimental results in a self-constructed floating waste dataset demonstrate that, compared to baseline YOLOv8s, the ES-YOLOv8 algorithm improves mAP@0.5 and mAP@0.5:0.95 by 5.4% and 6.1%, respectively. Compar ative experiments with state-of-the-art models further validate its superiority and effectiveness. Furthermore, experiments conducted on public datasets confirm the robustness and generalizability of ES-YOLOv8. This study aims to provide a high-precision, low-power-consumption technological solution for intelligent water governance, offering potential ecological and engineering applications.
为应对水面漂浮垃圾检测中的挑战,如小目标尺度、不规则形状和强背景干扰,本研究提出了一种基于YOLOv8s框架的增强检测模型,名为ES - YOLOv8。新模型优化了颈部的特征融合策略,构建了精细的“160 - 80 - 40 - 20”多尺度检测框架。与高效多尺度注意力(EMA)模块相结合,显著提高了模型提取小漂浮物体特征的能力。此外,采用了创新的Shape - IoU损失函数,通过形状敏感约束优化不规则目标的边界框回归精度。这导致开发出一种集成了特征增强、干扰抑制和定位优化的增强模型。在自建的漂浮垃圾数据集上的实验结果表明,与基线YOLOv8s相比,ES - YOLOv8算法的mAP@0.5和mAP@0.5:0.95分别提高了5.4%和6.1%。与现有先进模型的对比实验进一步验证了其优越性和有效性。此外,在公共数据集上进行的实验证实了ES - YOLOv8的鲁棒性和通用性。本研究旨在为智能水治理提供一种高精度、低功耗的技术解决方案,具有潜在的生态和工程应用价值。