基于改进YOLOv8的智能船舶水面多目标检测方法

Water surface multi-target detection method of intelligent ship water surface based on improved YOLOv8

  • 摘要: 【目的】为提升智能船舶在自动检测水面漂浮物体的稳定性与精确度,【方法】提出了一种基于改进YOLOv8算法的只能船舶水面多目标视觉检测方法。为增强模型对于不同尺度物体的检测能力,设计DDCNv4模块以替代骨干网络Backbone中的C2f模块。在与Neck层的连接处,引入Bi-Level Routing Attention注意力机制,以强化模型在抗干扰和纹理特征提取方面的性能,优化对小尺寸物体的检测效果。此外,在Neck层进一步集成了DSConv模块,以实现模型的轻量化,兼顾了模型性能与计算效率。【结果】实验结果显示,相较于传统的YOLOv5、YOLOv7和YOLOv8等方法,本研究提出改进YOLOv8算法的检测准确率达到90.4%,比原YOLOv8算法的mAP@0.5提升了5.6%,且参数量减少了0.1M,模型推理速度提高了1.98FPS。【结论】该算法在复杂多变的水面环境中,能够保持高水平的检测性能,为目标检测在智能船舶领域的应用提供理论参考。

     

    Abstract: Objectives In order to improve the stability and accuracy of intelligent ships in the automatic detection of floating objects on the water surface Methods a water surface multi-target detection method of intelligent ship based on improved YOLOv8 algorithm is proposed. In order to enhance the detection ability of the model for objects of different scales, DDCNv4 module is designed to replace C2f module in Backbone network. At the connection with the Neck layer, the Bi-Level Routing Attention mechanism is introduced to strengthen the performance of the model in anti-interference and texture feature extraction, and optimize the detection effect of small-size objects. In addition, the DSConv module is further integrated in the Neck layer to realize the lightweight of the model, which takes into account the model performance and computational efficiency. Results The experimental results show that compared with the traditional YOLOv5, YOLOv7, YOLOv8 and other methods, the detection accuracy of the improved YOLOv8 algorithm proposed in this study reaches 90.4%, which is 5.6% higher than the mAP@0.5 of the original YOLOv8 algorithm, and the number of parameters is reduced by 0.1M. The model inference speed is improved by 1.98FPS. Conclusions The algorithm can maintain a high level of detection performance in the complex water surface environment, which provides theoretical reference for the application of target detection in the field of intelligent ship.

     

/

返回文章
返回