基于改进YOLOv8的水下垃圾检测研究

Research on underwater garbage detection based on improved YOLOv8

  • 摘要: 针对水下环境光线昏暗、图像分辨率偏低和小目标密集等问题导致现有目标检测算法性能受限的核心痛点,提出一种改进的YOLOv8水下垃圾目标检测算法。该算法在YOLOv8的特征融合模块中引入了自校准卷积,采用分组卷积的方式进行多尺度特征提取,通过下采样操作来扩大网络的感受野,旨在提升模型的多尺度特征融合与检测能力,更准确地识别水下垃圾。在Seaclear Marine Debris数据集与TrashCan数据集上开展实验,相较于YOLOv8,改进后的模型在Seaclear Marine Debris数据集上的检测精度提升了1.5个百分点,mAP值提升了1.1个百分点;在TrashCan数据集上,检测精度提升了2.4个百分点,mAP值提升了0.7个百分点。实验结果表明,所提方法在水下复杂环境中仍能保持较高检测精度,满足水下垃圾检测的实际需求。

     

    Abstract: To address the core challenge that the existing object detection algorithms were constrained by issues such as low light, low image resolution, and dense small objects in underwater environments, an improved YOLOv8 algorithm for underwater garbage object detection was proposed. The algorithm introduced self‑calibrated convolution into the feature fusion module of YOLOv8, used the method of grouping convolution for multi scale feature extraction, and expanded the receptive field of the network through down-sampling operation, to improve the multi scale feature fusion and detection ability of the model, and more accurately identify underwater garbage. The experiments were carried out on the Seaclear Marine Debris data set and TrashCan data set. Compared with YOLOv8, the detection accuracy of the improved model on the Seaclear Marine Debris data set was improved by 1.5 percent point, mAP values increased by 1.1 percent point. On the TrashCan dataset, the detection accuracy was increased by 2.4 percent point, and the mAP value was increased by 0.7 percent point. Experimental results showed that the proposed method could maintain high detection accuracy in complex underwater environment, and could meet the actual needs of underwater garbage detection.

     

/

返回文章
返回