A Lightweight underwater detector enhanced by Attention mechanism, GSConv and WIoU on YOLOv8
- PMID: 39468159
- PMCID: PMC11519365
- DOI: 10.1038/s41598-024-75809-z
A Lightweight underwater detector enhanced by Attention mechanism, GSConv and WIoU on YOLOv8
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.
© 2024. The Author(s).
Conflict of interest statement
The authors declare no competing interests.
Figures
Similar articles
-
FEB-YOLOv8: A multi-scale lightweight detection model for underwater object detection.PLoS One. 2024 Sep 27;19(9):e0311173. doi: 10.1371/journal.pone.0311173. eCollection 2024. PLoS One. 2024. PMID: 39331686 Free PMC article.
-
RSE-YOLOv8: An Algorithm for Underwater Biological Target Detection.Sensors (Basel). 2024 Sep 18;24(18):6030. doi: 10.3390/s24186030. Sensors (Basel). 2024. PMID: 39338779 Free PMC article.
-
YOLOv8-MU: An Improved YOLOv8 Underwater Detector Based on a Large Kernel Block and a Multi-Branch Reparameterization Module.Sensors (Basel). 2024 May 1;24(9):2905. doi: 10.3390/s24092905. Sensors (Basel). 2024. PMID: 38733010 Free PMC article.
-
Lightweight enhanced YOLOv8n underwater object detection network for low light environments.Sci Rep. 2024 Nov 13;14(1):27922. doi: 10.1038/s41598-024-79211-7. Sci Rep. 2024. PMID: 39537721 Free PMC article.
-
Transmission Line Defect Target-Detection Method Based on GR-YOLOv8.Sensors (Basel). 2024 Oct 24;24(21):6838. doi: 10.3390/s24216838. Sensors (Basel). 2024. PMID: 39517735 Free PMC article.
References
-
- Zou Z, Chen K, Shi Z, et al. Object detection in 20 years: A survey[J]. Proceedings of the IEEE, (2023).
-
- Fu, H., Song, G. & Wang, Y. Improved YOLOv4 marine target detection combined with CBAM. Symmetry 13(4), 623 (2021).
-
- Liu, Y. et al. Ocean explorations using autonomy: Technologies, strategies and applications[C]//Offshore Robotics: I(1). Springer Singapore 2022, 35–58 (2021).
-
- Yuh, J. & West, M. Underwater robotics. Adv. Robot. 15(5), 609–639 (2001).
-
- 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
Full Text Sources