LU2Net: A Lightweight Network for Real-time Underwater Image Enhancement

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Yang, Haodong, Xu, Jisheng, Lin, Zhiliang, He, Jianping
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910497291370496
author Yang, Haodong
Xu, Jisheng
Lin, Zhiliang
He, Jianping
author_facet Yang, Haodong
Xu, Jisheng
Lin, Zhiliang
He, Jianping
contents Computer vision techniques have empowered underwater robots to effectively undertake a multitude of tasks, including object tracking and path planning. However, underwater optical factors like light refraction and absorption present challenges to underwater vision, which cause degradation of underwater images. A variety of underwater image enhancement methods have been proposed to improve the effectiveness of underwater vision perception. Nevertheless, for real-time vision tasks on underwater robots, it is necessary to overcome the challenges associated with algorithmic efficiency and real-time capabilities. In this paper, we introduce Lightweight Underwater Unet (LU2Net), a novel U-shape network designed specifically for real-time enhancement of underwater images. The proposed model incorporates axial depthwise convolution and the channel attention module, enabling it to significantly reduce computational demands and model parameters, thereby improving processing speed. The extensive experiments conducted on the dataset and real-world underwater robots demonstrate the exceptional performance and speed of proposed model. It is capable of providing well-enhanced underwater images at a speed 8 times faster than the current state-of-the-art underwater image enhancement method. Moreover, LU2Net is able to handle real-time underwater video enhancement.
format Preprint
id arxiv_https___arxiv_org_abs_2406_14973
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LU2Net: A Lightweight Network for Real-time Underwater Image Enhancement
Yang, Haodong
Xu, Jisheng
Lin, Zhiliang
He, Jianping
Computer Vision and Pattern Recognition
Image and Video Processing
Computer vision techniques have empowered underwater robots to effectively undertake a multitude of tasks, including object tracking and path planning. However, underwater optical factors like light refraction and absorption present challenges to underwater vision, which cause degradation of underwater images. A variety of underwater image enhancement methods have been proposed to improve the effectiveness of underwater vision perception. Nevertheless, for real-time vision tasks on underwater robots, it is necessary to overcome the challenges associated with algorithmic efficiency and real-time capabilities. In this paper, we introduce Lightweight Underwater Unet (LU2Net), a novel U-shape network designed specifically for real-time enhancement of underwater images. The proposed model incorporates axial depthwise convolution and the channel attention module, enabling it to significantly reduce computational demands and model parameters, thereby improving processing speed. The extensive experiments conducted on the dataset and real-world underwater robots demonstrate the exceptional performance and speed of proposed model. It is capable of providing well-enhanced underwater images at a speed 8 times faster than the current state-of-the-art underwater image enhancement method. Moreover, LU2Net is able to handle real-time underwater video enhancement.
title LU2Net: A Lightweight Network for Real-time Underwater Image Enhancement
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2406.14973