YOLOv8-C2f-Faster-EMA: An Improved Underwater Trash Detection Model Based on YOLOv8

Jin Zhu,Tao Hu,Linhan Zheng,Nan Zhou,Huilin Ge,Zhichao Hong
DOI: https://doi.org/10.3390/s24082483
IF: 3.9
2024-04-13
Sensors
Abstract:Anthropogenic waste deposition in aquatic environments precipitates a decline in water quality, engendering pollution that adversely impacts human health, ecological integrity, and economic endeavors. The evolution of underwater robotic technologies heralds a new era in the timely identification and extraction of submerged litter, offering a proactive measure against the scourge of water pollution. This study introduces a refined YOLOv8-based algorithm tailored for the enhanced detection of small-scale underwater debris, aiming to mitigate the prevalent challenges of high miss and false detection rates in aquatic settings. The research presents the YOLOv8-C2f-Faster-EMA algorithm, which optimizes the backbone, neck layer, and C2f module for underwater characteristics and incorporates an effective attention mechanism. This algorithm improves the accuracy of underwater litter detection while simplifying the computational model. Empirical evidence underscores the superiority of this method over the conventional YOLOv8n framework, manifesting in a significant uplift in detection performance. Notably, the proposed method realized a 6.7% increase in precision (P), a 4.1% surge in recall (R), and a 5% enhancement in mean average precision (mAP). Transcending its foundational utility in marine conservation, this methodology harbors potential for subsequent integration into remote sensing ventures. Such an adaptation could substantially enhance the precision of detection models, particularly in the realm of localized surveillance, thereby broadening the scope of its applicability and impact.
engineering, electrical & electronic,chemistry, analytical,instruments & instrumentation
What problem does this paper attempt to address?
The paper mainly addresses the following issues: 1. **Background & Challenges**: The paper begins by discussing the importance and urgency of trash detection in underwater environments, especially for small-sized debris. The challenges faced by current methods include high false positive and false negative rates. 2. **Objectives & Solutions**: The research aims to propose an improved underwater trash detection model based on YOLOv8 — YOLOv8-C2f-Faster-EMA, to enhance the detection accuracy of small-sized debris. The model enhances feature extraction capabilities by optimizing the backbone network, neck layers, and introducing an effective attention mechanism, while simplifying the computational model. 3. **Specific Improvements**: - Introduction of the C2f-Faster-EMA module to enhance multi-scale feature fusion and attention mechanisms, improving the precision and robustness of object detection. - Integration of FasterNet and Efficient Multi-Scale Attention (EMA) modules, significantly increasing the Average Precision (AP). - Verification of the superior performance of the proposed method through a series of benchmark tests. In summary, the paper proposes an improved algorithm based on YOLOv8 for existing underwater trash detection technologies that struggle with detecting small-sized targets. The aim is to improve detection accuracy, reduce false positives and negatives, thereby providing a more effective method for underwater environmental monitoring.