A Lightweight Underwater Object Detection Algorithm with Adaptive Image Enhancement Based on YOLOv8

Zuxin Zhao*, Jiarong Han, Zhongjing Ma, Suli Zou, Yu Liu, Guancheng Li

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

This paper introduces a lightweight underwater object detection algorithm based on YOLOv8, essential for underwater robotics challenged by environmental complexity and real-time demands. Firstly, to enhance underwater image quality without significantly increasing computational demands, this study introduces an Adaptive Underwater Image Enhancement module utilizing lightweight convolutions and digital filters for dynamic enhancement. Secondly, a Re-parameterized Partial Convolution Block is proposed and integrated, replacing foundational blocks in the baseline model's architecture, resulting in reduced detection network parameters and enhanced accuracy. Additionally, performance evaluation on the UTDAC dataset demonstrates our model achieving a 46.8%mAP, marking a 1.6% improvement over the baseline, with a total parameter count of merely 2.81 M. Ablation studies and extended experiments validate the effectiveness and adaptability of the proposed modules. Experimental results show that the model achieves a superior balance between accuracy and processing speed, making it particularly suitable for underwater robotic perception.

Original languageEnglish
Title of host publicationProceedings of the 43rd Chinese Control Conference, CCC 2024
EditorsJing Na, Jian Sun
PublisherIEEE Computer Society
Pages7830-7835
Number of pages6
ISBN (Electronic)9789887581581
DOIs
Publication statusPublished - 2024
Event43rd Chinese Control Conference, CCC 2024 - Kunming, China
Duration: 28 Jul 202431 Jul 2024

Publication series

NameChinese Control Conference, CCC
ISSN (Print)1934-1768
ISSN (Electronic)2161-2927

Conference

Conference43rd Chinese Control Conference, CCC 2024
Country/TerritoryChina
CityKunming
Period28/07/2431/07/24

Keywords

  • Structural Re-parameterization
  • Underwater Image Enhancement
  • Underwater Object Detection
  • YOLOv8

Fingerprint

Dive into the research topics of 'A Lightweight Underwater Object Detection Algorithm with Adaptive Image Enhancement Based on YOLOv8'. Together they form a unique fingerprint.

Cite this