Laser & Optoelectronics Progress, Volume. 62, Issue 4, 0437008(2025)

Lightweight Underwater Optical Image Recognition Algorithm Based on YOLOv8

Shun Cheng1,2、*, Jianrong Li1, Zhiqian Wang1, Shaojin Liu1, and Muyuan Wang1,2
Author Affiliations
  • 1Changchun Institute of Optics, Fine Mechanics and Physics, Chinese Academy of Sciences, Changchun 130033, Jilin , China
  • 2Daheng College, University of Chinese Academy of Sciences, Beijing 100049, China
  • show less

    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.

    Keywords
    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Digital Image Processing

    Received: May. 17, 2024

    Accepted: Jul. 29, 2024

    Published Online: Feb. 10, 2025

    The Author Email:

    DOI:10.3788/LOP241313

    CSTR:32186.14.LOP241313

    Topics