TY - GEN
T1 - A Lightweight Underwater Object Detection Algorithm with Adaptive Image Enhancement Based on YOLOv8
AU - Zhao, Zuxin
AU - Han, Jiarong
AU - Ma, Zhongjing
AU - Zou, Suli
AU - Liu, Yu
AU - Li, Guancheng
N1 - Publisher Copyright:
© 2024 Technical Committee on Control Theory, Chinese Association of Automation.
PY - 2024
Y1 - 2024
N2 - 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.
AB - 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.
KW - Structural Re-parameterization
KW - Underwater Image Enhancement
KW - Underwater Object Detection
KW - YOLOv8
UR - http://www.scopus.com/inward/record.url?scp=85205511058&partnerID=8YFLogxK
U2 - 10.23919/CCC63176.2024.10661480
DO - 10.23919/CCC63176.2024.10661480
M3 - Conference contribution
AN - SCOPUS:85205511058
T3 - Chinese Control Conference, CCC
SP - 7830
EP - 7835
BT - Proceedings of the 43rd Chinese Control Conference, CCC 2024
A2 - Na, Jing
A2 - Sun, Jian
PB - IEEE Computer Society
T2 - 43rd Chinese Control Conference, CCC 2024
Y2 - 28 July 2024 through 31 July 2024
ER -