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

Abstract

Abstract The underwater target detection is the most important part of monitoring for environment, ocean, and other fields. However, the detection accuracy is greatly decreased by the poor image quality resulted from the complex underwater environments. The storage and computing power of underwater equipments are not enough for complex underwater target detection technology. Therefore, many YOLO series algorithms have been applied to underwater target detection. On the basis of YOLOv8, a lightweight underwater detector enhanced by Attention mechanism, GSConv and WIoU, AGW-YOLOv8, is proposed in this paper. Firstly, by the combination of limited contrast adaptive histogram equalization and wavelet transform(LCAHE-WT), the fidelity and detail of images are improved; Secondly, by CBAM, the key channel features can be effectively extracted with retaining spatial information to improve the performance of the network when dealing with complex image tasks; Thirdly, by GSConv, composed of depth-wise separable convolution and regular convolution, the model parameters and computational complexity are reduced; Fourth, the SE attention mechanism is integrated into the C2f module of the neck, and the channel dimension is weighted to make the network more focus on important features, and to further enhance the feature extraction capability; Finally, by the dynamic nonmonotonic mechanism of WIoU, the gradient gain can be reasonably distributed, the harmful gradients of extreme samples can be reduced, and the generalization ability and overall performance of the model are improved. By the experiments on the URPC2020 data-set, it can been proved that the mAP of AGW-YOLOv8 can reaches 82.9%, is 2.5% higher than that of YOLOv8; and the parameters is 2.95M, is lower than 3.01M of YOLOv8.

Keywords

• Underwater target detection
• Lightweight
• Attention mechanism
• Image enhancement