A method for detecting dead fish on large water surfaces based on improved YOLOv10

Qingbin Tian,Yukang Huo,Mingyuan Yao,Haihua Wang
2024-08-31
Abstract:Dead fish frequently appear on the water surface due to various factors. If not promptly detected and removed, these dead fish can cause significant issues such as water quality deterioration, ecosystem damage, and disease transmission. Consequently, it is imperative to develop rapid and effective detection methods to mitigate these challenges. Conventional methods for detecting dead fish are often constrained by manpower and time limitations, struggling to effectively manage the intricacies of aquatic environments. This paper proposes an end-to-end detection model built upon an enhanced YOLOv10 framework, designed specifically to swiftly and precisely detect deceased fish across extensive water surfaces.Key enhancements include: (1) Replacing YOLOv10's backbone network with FasterNet to reduce model complexity while maintaining high detection accuracy; (2) Improving feature fusion in the Neck section through enhanced connectivity methods and replacing the original C2f module with CSPStage modules; (3) Adding a compact target detection head to enhance the detection performance of smaller objects. Experimental results demonstrate significant improvements in P(precision), R(recall), and AP(average precision) compared to the baseline model YOLOv10n. Furthermore, our model outperforms other models in the YOLO series by significantly reducing model size and parameter count, while sustaining high inference speed and achieving optimal AP performance. The model facilitates rapid and accurate detection of dead fish in large-scale aquaculture systems. Finally, through ablation experiments, we systematically analyze and assess the contribution of each model component to the overall system performance.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve This paper aims to address the problem of quickly and accurately detecting dead fish on a large-scale water surface. Specifically: 1. **Background and Challenges**: - The aquaculture industry is rapidly developing worldwide, with China being the largest producer and exporter of aquatic products. - The presence of dead fish can lead to water pollution and disease spread, necessitating timely detection and removal. - Traditional methods (such as manual observation) are time-consuming and inefficient, making it difficult to effectively address issues in complex water environments. 2. **Proposed Method**: - A detection model based on an improved YOLOv10 framework is proposed for quickly and accurately detecting dead fish on large-scale water surfaces. - The main improvements include: - Replacing YOLOv10's backbone network with FasterNet to reduce model complexity while maintaining high accuracy. - Enhancing feature fusion in the Neck part through improved connection methods and replacing the original C2f module with the CSPStage module. - Adding a compact object detection head to improve the detection performance of small targets. 3. **Experimental Results**: - Experimental results show significant improvements in precision (P), recall (R), and average precision (AP) compared to the baseline model YOLOv10n. - The model maintains high inference speed while reducing model size and the number of parameters, achieving the best AP performance. - Ablation experiments systematically evaluated the contribution of each model component to the overall system performance. In summary, the main objective of this paper is to develop an efficient and accurate dead fish detection method to support the sustainable development of large-scale aquaculture systems.