Heliyon (May 2024)
Shallow mud detection algorithm for submarine channels based on improved YOLOv5s
Abstract
Submarine mud poses a risk to channel navigation safety. Traditional detection methods lack efficiency and accuracy. As a result, this paper proposed an enhanced shallow submarine mud detection algorithm, leveraging an improved YOLOv5s model to increase accuracy and effectiveness in identifying such hazards in marine channels. Firstly, the sub-bottom profiler was employed to assess the submarine channel of Lianyungang Port to acquire the image data of the shallow mud sound print. Concurrently, the analysis incorporated the characteristics of changes in sound intensity peaks to precisely identify the shallow mud's location. Furthermore, the incorporation of C2F feature module into the backbone module enhances the gradient flow of the algorithm, augments the feature extraction information, and improves the algorithm's detection performance. Subsequently, Efficient Multi-Scale Attention (EMA) mechanism is incorporated into the neck module, aiming to optimize the algorithm's channel dimensions, minimize computational overhead, and enhance its detection efficiency. Finally, the study introduced Normalized Wasserstein Distance (NWD) loss function into bounding box regression loss function. This integration effectively addresses the issue of multi-scale defects, emphasizes the transformation of target planar position deviation, and improves the accuracy of the algorithm's detection capabilities. The results indicate that the improved YOLOv5s-EF algorithm outperforms the original YOLOv5s algorithm and other widely used detection algorithms. It achieved a validation set precision rate of 97.8%, recall rate of 97.6%, F1 value of 97.7%, mean Average Precision (mAP)@0.5 of 98.2%, [email protected] of 69.6%, and Frames Per Second (FPS) of 51.8. YOLOv5s-EF algorithm proposed in this study offers a novel technical approach for detecting mud in submarine channels, which is importance for ensuring the safe operation and maintenance of dredging in such channels.
