搜索

x

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

An improved YOLOv5s recognition and detection algorithm for floating objects on lake surface

  • 摘要: 为解决YOLOv5s算法在湖面场景下对漂浮物误检和漏检的问题,提出一种改进的轻量化YOLOv5s算法,以提高对湖面漂浮物的识别。改进的轻量化YOLOv5s算法在Backbone层采用轻量化ShufflenetV2_cssp网络并结合DSPPF_CS模块,在Neck层引入改进RFBSD模块,同时通过CIoU_SC损失函数与尺度缩放机制优化边界框回归。实验结果表明,改进的轻量化YOLOv5s算法能有效降低湖面检测中的误检率和漏检率,特别是在强反光场景下可抑制水面波纹误判,漏检率显著降低;在保障检测精度前提下,实现了帧率速度的优化,为湖面漂浮物检测应用提供支持。

     

    Abstract: To address the question of issue of misdetection and missed detection on the YOLOv5s algorithm for floating objects in lake surface scenarios, an improved lightweight YOLOv5s algorithm is proposed to improve detection of lake surface floating objects. In the Backbone layer, the improved lightweight ShufflenetV2_cssp network is adopted combined with the DSPPF_CS module. In the Neck layer, an improved RFBSD module is introduced, and meanwhile, the CIoU_SC loss function and the scale scaling mechanism are employed to optimize bounding box regression. Experimental results demonstrate that the improved lightweight YOLOv5s algorithm can effectively mitigate the false detection and missed detection issues in lake surface detection, especially can suppress the misjudgment of water surface ripples in strong reflection scenarios, and the missed detection rate is significantly reduced. While ensuring detection accuracy, the algorithm achieves the optimization of frame rate, providing technical support for the practical application of lake surface floating object detection.

     

/

返回文章
返回