Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2024 Oct 28;14(1):25797.
doi: 10.1038/s41598-024-75809-z.

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

Affiliations

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

Shaobin Cai et al. Sci Rep. .

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: Attention mechanism; Image enhancement; Lightweight; Underwater target detection.

PubMed Disclaimer

Conflict of interest statement

The authors declare no competing interests.

Figures

Fig. 1
Fig. 1
Sample images of URPC2020.
Fig. 2
Fig. 2
Flip, Zoom, Crop.
Fig. 3
Fig. 3
Mosaic enhancement.
Fig. 4
Fig. 4
Haar Wavelet Pool.
Fig. 5
Fig. 5
The result of wavelet transform and LCAHE-WT.
Fig. 6
Fig. 6
CBAM structure diagram.
Fig. 7
Fig. 7
GSConv module structure diagram.
Fig. 8
Fig. 8
depth-wise separable convolution module structure diagram.
Fig. 9
Fig. 9
point-wise convolution module structure diagram.
Fig. 10
Fig. 10
SE attention mechanism structure diagram.
Fig. 11
Fig. 11
SE-C2f module structure diagram.
Fig. 12
Fig. 12
Block_SE module structure diagram.
Fig. 13
Fig. 13
Entity Diagram of Loss Function.
Fig. 14
Fig. 14
AGW-YOLOv8 model diagram.
Fig. 15
Fig. 15
PR curve of the proposed model.
Fig. 16
Fig. 16
Heat map.
Fig. 17
Fig. 17
Comparison of the results of two models for small target detection.
Fig. 18
Fig. 18
Comparison of the results of the two models for the detection of complex terrain.
Fig. 19
Fig. 19
Comparison of the results of the two models for the detection of dark water.

Similar articles

References

    1. Zou Z, Chen K, Shi Z, et al. Object detection in 20 years: A survey[J]. Proceedings of the IEEE, (2023).
    1. Fu, H., Song, G. & Wang, Y. Improved YOLOv4 marine target detection combined with CBAM. Symmetry 13(4), 623 (2021).
    1. Liu, Y. et al. Ocean explorations using autonomy: Technologies, strategies and applications[C]//Offshore Robotics: I(1). Springer Singapore 2022, 35–58 (2021).
    1. Yuh, J. & West, M. Underwater robotics. Adv. Robot. 15(5), 609–639 (2001).
    1. Liu, C. et al. A new dataset, Poisson GAN and AquaNet for underwater object grabbing. IEEE Trans. Circuits Syst. Video Technol. 32(5), 2831–2844 (2021).

LinkOut - more resources