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.

Research on an underwater image segmentation algorithm based on YOLOv8

Research on Underwater Small Target Detection Algorithm Based on Improved YOLOv3

YoloXT: A Object Detection Algorithm for Marine Benthos

Underwater Object Detection Based on Enhanced YOLO

RSE-YOLOv8: An Algorithm for Underwater Biological Target Detection

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

Underwater Robot Target Detection Algorithm Based on YOLOv8

Underwater Object Detection Algorithm Based on an Improved YOLOv8

Underwater target detection based on improved YOLOv7

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

Underwater target detection algorithm based on improved YOLOv4 with SemiDSConv and FIoU loss function

Attention-Based Lightweight YOLOv8 Underwater Target Recognition Algorithm

Enhanced YOLOv7 with three-dimensional attention and its application into underwater object detection

Underwater-YCC: Underwater Target Detection Optimization Algorithm Based on YOLOv7

An underwater target recognition method based on improved YOLOv4 in complex marine environment

Underwater Target Detection Algorithm Based on Improved YOLOv5

Instance Segmentation of Underwater Images by Using Deep Learning

A Lightweight underwater detector enhanced by Attention mechanism, GSConv and WIoU on YOLOv8

Underwater Target Recognition Based on Improved YOLOv4 Neural Network

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

Research on underwater object recognition based on YOLOv3