As the problem of marine debris worsens, underwater garbage detection is of paramount importance for the protection and restoration of marine ecosystems. Due to the complexity of the underwater environment and the blurring of underwater light, the contours of detection targets are unclear, resulting in poor imaging quality and difficulties in distinguishing detection targets. This affects the accuracy and precision of traditional target detection algorithms in underwater garbage detection. After research, this paper proposes a new target detection algorithm called YOLO-PSI. Based on YOLOv8n, a new module called FDP is proposed. The FDP module, through a customized feature focusing module and a feature diffusion mechanism, enables features at each scale to contain detailed contextual information, enhancing feature extraction capabilities. The C2f-PPA module is introduced into the network backbone, mainly for detecting small underwater targets. Through parallel feature extraction, it improves the efficiency of feature extraction and enhances the detection capability of small underwater targets. Focal Modulation replaces the SPPF module in the model to reduce unnecessary computation. The Focal Modulation module increases attention to multi-scale feature maps through weighted fusion, addressing the problem of multi-scale feature map fusion. Experimental results show that the improved YOLOv8-PSI model achieves a mean average precision (mAP@0.5) of 70.3%, which is 3.8% higher than the original YOLOv8n model, making it more suitable for underwater target detection in complex environments.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Underwater Target Detection Improved Algorithm Based on Deep Learning

  • Chaojie Ma,
  • Qilong Sun,
  • Zhihao Jiang

摘要

As the problem of marine debris worsens, underwater garbage detection is of paramount importance for the protection and restoration of marine ecosystems. Due to the complexity of the underwater environment and the blurring of underwater light, the contours of detection targets are unclear, resulting in poor imaging quality and difficulties in distinguishing detection targets. This affects the accuracy and precision of traditional target detection algorithms in underwater garbage detection. After research, this paper proposes a new target detection algorithm called YOLO-PSI. Based on YOLOv8n, a new module called FDP is proposed. The FDP module, through a customized feature focusing module and a feature diffusion mechanism, enables features at each scale to contain detailed contextual information, enhancing feature extraction capabilities. The C2f-PPA module is introduced into the network backbone, mainly for detecting small underwater targets. Through parallel feature extraction, it improves the efficiency of feature extraction and enhances the detection capability of small underwater targets. Focal Modulation replaces the SPPF module in the model to reduce unnecessary computation. The Focal Modulation module increases attention to multi-scale feature maps through weighted fusion, addressing the problem of multi-scale feature map fusion. Experimental results show that the improved YOLOv8-PSI model achieves a mean average precision (mAP@0.5) of 70.3%, which is 3.8% higher than the original YOLOv8n model, making it more suitable for underwater target detection in complex environments.