Dead Fish Detection Model Based on DD-IYOLOv8

Abstract

In aquaculture, the presence of dead fish on the water surface can serve as a bioindicator of health issues or environmental stressors. To enhance the precision of detecting dead fish floating on the water’s surface, this paper proposes a detection approach that integrates data-driven insights with advanced modeling techniques. Firstly, to reduce the influence of aquatic disturbances and branches during the identification process, prior information, such as branches and ripples, is annotated in the dataset to guide the model to better learn the scale and shape characteristics of dead fish, reduce the interference of branch ripples on detection, and thus improve the accuracy of target identification. Secondly, leveraging the foundational YOLOv8 architecture, a DD-IYOLOv8 (Data-Driven Improved YOLOv8) dead fish detection model is designed. Considering the significant changes in the scale of dead fish at different distances, DySnakeConv (Dynamic Snake Convolution) is introduced into the neck network detection head to adaptively adjust the receptive field, thereby improving the network’s capability to capture features. Additionally, a layer for detecting minor objects has been added, and the detection head of YOLOv8 has been modified to 4, allowing the network to better focus on small targets and occluded dead fish, which improves detection performance. Furthermore, the model incorporates a HAM (Hybrid Attention Mechanism) in the later stages of the backbone network to refine global feature extraction, sharpening the model’s focus on dead fish targets and further enhancing detection accuracy. The experimental results showed that the accuracy of DD-IYOLOv8 in detecting dead fish reached 92.8%, the recall rate reached 89.4%, the AP reached 91.7%, and the F1 value reached 91.0%. This study can achieve precise identification of dead fish, which will help promote the research of automatic pond patrol machine ships.