Underwater Image Enhancement Algorithm Based on Adversarial Training

Monan Zhang,Yichen Li,Wenbin Yu

DOI: https://doi.org/10.3390/electronics13112184

IF: 2.9

2024-06-04

Electronics

Abstract:Ocean observation is the first step in the development of the ocean, whose abundant resources and strategic significance are attracting increasing attention. Observation methods based on visual sensor networks have received great attention from researchers due to their visualization capability and high information capacity. However, below the sea surface, objective factors such as blurriness, turbulence, and underwater color casting can cause image distortion and affect the acquisition of images. In this paper, the enhancement of underwater images is tackled using an adversarial learning-based approach. First, pre-processing is applied to address the significant color casting in the dataset, thus enhancing feature learning for subsequent style transfer. Then, corresponding improvements are made to a generative adversarial network's structure and loss functions to better restore the features of the network output. Finally, evaluations and comparisons are performed using underwater image quality assessment metrics and several public datasets. Through multidimensional experiments, the proposed algorithm is shown to exhibit excellent performance in both subjective and objective evaluation metrics compared to state-of-the-art algorithms, as well as in practical visual applications.

engineering, electrical & electronic,computer science, information systems,physics, applied

What problem does this paper attempt to address?

The problem this paper attempts to address is the color distortion and blurring issues in Underwater Image Enhancement (UIE). Specifically, objective factors in underwater imaging environments such as blurring, turbulence, and underwater color shift can lead to image distortion, affecting the quality of image acquisition. Traditional image enhancement algorithms have limited effectiveness in these harsh environments, thus necessitating the development of image enhancement techniques specifically tailored for underwater scenarios.

To tackle these challenges, the paper proposes an underwater image enhancement algorithm based on adversarial training. The main methods include:

1. **Preprocessing**: Preprocessing the dataset through color balancing and fusion methods to optimize feature learning, thereby improving the effectiveness of subsequent style transfer.
2. **Network Structure and Loss Function Improvement**: Improving the structure and loss function of the Generative Adversarial Network (GAN) to better restore the features of the network output.
3. **Evaluation and Comparison**: Using underwater image quality evaluation metrics and multiple public datasets for evaluation and comparison to verify the effectiveness of the algorithm.

Through multi-dimensional experiments, the algorithm demonstrates excellent performance in both subjective and objective evaluation metrics and shows good performance in practical visual applications.