Laser & Optoelectronics Progress, Volume. 62, Issue 4, 0437008(2025)
Lightweight Underwater Optical Image Recognition Algorithm Based on YOLOv8
To address the challenges of low recognition accuracy and high computational complexity in underwater optical target recognition algorithms, a lightweight YOLOv8 underwater optical recognition algorithm based on automatic color equalization (ACE) image enhancement is proposed. Initially, we apply the ACE image enhancement algorithm to preprocess images. Subsequently, we improve the feature extraction capabilities by replacing the YOLOv8 backbone with an upgraded SENetV2 backbone network. To further decrease computational quantity, we introduce a lightweight cross-scale feature fusion module in place of the neck network. Then, we utilize DySample as a substitute for the traditional upsampler to improve image processing efficiency. We refine the DyHead detection head to better perceive targets. Finally, we enhance the accuracy of bounding box regression by replacing loss function of YOLOv8 with InnerMPDIoU based on the minimum point distance intersection ratio (MPDIoU). Experimental results show that the proposed SCDDI-YOLOv8 algorithm achieves a mean average precision of 77.3% and 71.5% on the URPC2020 and UWG datasets, respectively, while reducing parameters by ~20.7%, floating-point operations by 6×108, and model size by 1.2 MB compared with the original YOLOv8n. Compared with other advanced algorithms, the proposed algorithm can meet the sensitive computational needs of edge devices.
Get Citation
Copy Citation Text
Shun Cheng, Jianrong Li, Zhiqian Wang, Shaojin Liu, Muyuan Wang. Lightweight Underwater Optical Image Recognition Algorithm Based on YOLOv8[J]. Laser & Optoelectronics Progress, 2025, 62(4): 0437008
Category: Digital Image Processing
Received: May. 17, 2024
Accepted: Jul. 29, 2024
Published Online: Feb. 10, 2025
The Author Email:
CSTR:32186.14.LOP241313