搜索

x

一种改进的YOLOv5s识别检测湖面漂浮物算法

An Improved YOLOv5s Recognition and Detection Algorithm for Floating Objects on Lake Surface

  • 摘要: 为解决YOLOv5s算法的算法难以满足嵌入式系统需求的问题,提出一种改进的YOLOv5s轻量化检测算法,实现实时的目标检测。首先改进的轻量化检测算法在Backbone层采用ShufflenetV2_cssp替换CSP-DarkNet,结合DSPPF_CS模块提升帧率与感受野,其次Neck层引入改进RFBSD模块优化参数分布,增强目标检测能力,并通过CIoU_SC损失函数与尺度缩放机制优化边界框回归,最后实验结果表明,参数量减少78.5%,帧率提升10帧·s−1,精度提高1.9%,展现轻量化与高性能优势,为嵌入式系统提供支持。

     

    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.

     

/

返回文章
返回