An Improved YOLOv5s Recognition and Detection Algorithm for Floating Objects on Lake Surface
-
Graphical Abstract
-
Abstract
To address the challenges traditional detection algorithms face in meeting the requirements of embedded systems, a modified lightweight YOLOv5s detection algorithm is proposed for real-time object detection. The proposed lightweight detection algorithm first replaces the CSP-DarkNet with ShufflenetV2_cssp in the Backbone layer, and combines the DSPPF_CS module to improve frame rate and receptive field. In addition, the Neck layer introduces the improved Receptive Field Block (RFBSD) structure to optimize parameter distribution and enhance object detection capability. The CIoU_SC loss function and scale scaling mechanism are then applied to optimize bounding box regression. Experimental results demonstrate that the number of parameters is reduced by 78.5%, the frame rate is increased by 10 fps, and accuracy improves by 1.9%. This showcases the advantages of lightweight design and high performance, providing support for embedded systems.
-
-