Abstract:Underwater target detection is widely used in various applications such as underwater search and rescue, underwater environment monitoring, and marine resource surveying. However, the complex underwater environment, including factors such as light changes and background noise, poses a significant challenge to target detection. We propose an improved underwater target detection algorithm based on YOLOv8n to overcome these problems. Our algorithm focuses on three aspects. Firstly, we replace the original C2f module with Deformable Convnets v2 to enhance the adaptive ability of the target region in the convolution check feature map and extract the target region’s features more accurately. Secondly, we introduce SimAm, a non-parametric attention mechanism, which can deduce and assign three-dimensional attention weights without adding network parameters. Lastly, we optimize the loss function by replacing the CIoU loss function with the Wise-IoU loss function. We named our new algorithm DSW-YOLOv8n, which is an acronym of Deformable Convnets v2, SimAm, and Wise-IoU of the improved YOLOv8n(DSW-YOLOv8n). To conduct our experiments, we created our own dataset of underwater target detection for experimentation. Meanwhile, we also utilized the Pascal VOC dataset to evaluate our approach. The mAP@0.5 and mAP@0.5:0.95 of the original YOLOv8n algorithm on underwater target detection were 88.6% and 51.8%, respectively, and the DSW-YOLOv8n algorithm mAP@0.5 and mAP@0.5:0.95 can reach 91.8% and 55.9%. The original YOLOv8n algorithm was 62.2% and 45.9% mAP@0.5 and mAP@0.5:0.95 on the Pascal VOC dataset, respectively. The DSW-YOLOv8n algorithm mAP@0.5 and mAP@0.5:0.95 were 65.7% and 48.3%, respectively. The number of parameters of the model is reduced by about 6%. The above experimental results prove the effectiveness of our method.

Improved YOLOv8 Algorithm for Water Surface Object Detection

MDM-YOLO: Research on Object Detection Algorithm Based on Improved YOLOv4 for Marine Organisms.

Research on Underwater Small Target Detection Algorithm Based on Improved YOLOv3

A Kind of Water Surface Multi-Scale Object Detection Method Based on Improved YOLOv5 Network

DSW-YOLOv8n: A New Underwater Target Detection Algorithm Based on Improved YOLOv8n

Detection of the Floating Objects on the Water Surface Based on Improved YOLOv5

Underwater Robot Target Detection Algorithm Based on YOLOv8

Underwater Object Detection Algorithm Based on an Improved YOLOv8

Attention-Based Lightweight YOLOv8 Underwater Target Recognition Algorithm

Soft-NMS-Enabled YOLOv5 with SIOU for Small Water Surface Floater Detection in UAV-Captured Images

YOLOv7-SN: Underwater Target Detection Algorithm Based on Improved YOLOv7

Enhanced Water Surface Object Detection with Dynamic Task-Aligned Sample Assignment and Attention Mechanisms

A YOLOv5s Algorithm Model Applied to Underwater Object Detection

Improved YOLOv7 model for underwater sonar image object detection

LWS-YOLOv7: A Lightweight Water-Surface Object-Detection Model

An Improved Underwater Target Detection Algorithm Based on YOLOX

Research on small target detection algorithm for water surface in complex weather environment

YOLOv8-MU: An Improved YOLOv8 Underwater Detector Based on a Large Kernel Block and a Multi-Branch Reparameterization Module

RSE-YOLOv8: An Algorithm for Underwater Biological Target Detection

Improved Underwater Transparent Organism Detection Based on YOLOv8

An Improved YOLOv5 Underwater Detector Based on an Attention Mechanism and Multi-Branch Reparameterization Module