Laser & Optoelectronics Progress, Volume. 58, Issue 12, 1210024(2021)
Underwater Image Enhancement Based on Generative Adversarial Network with Preprocessed Image Penalty
Aiming at mitigating the problems of low contrast, blurred details, and color distortion in underwater images, an underwater image enhancement method based on preprocessed image penalty and generative adversarial network (GAN) is proposed in this paper. First, an improved red channel histogram stretching algorithm is used to preprocess the input underwater image to improve the image contrast and avoid over enhancement of local blocks after traditional histogram stretching. Then, GAN with preprocessed image penalty is designed to realize underwater image enhancement. Moreover, multiscale convolution is used for the first two layers of the generator coding-decoding structure to enhance the detailed information learning ability of the network. Finally, a multiloss function is established in which the preprocessed image is used as a false truth value to impose loss penalty on GAN to improve generalization performance of the network. Experimental results show that compared with traditional image enhancement methods and deep learning-based image enhancement methods, the method performs better in terms of color deviation, contrast, and detailed information of underwater images, and has better robustness.
Get Citation
Copy Citation Text
Wei Song, Jingjing Xing, Yanling Du, Qi He. Underwater Image Enhancement Based on Generative Adversarial Network with Preprocessed Image Penalty[J]. Laser & Optoelectronics Progress, 2021, 58(12): 1210024
Category: Image Processing
Received: Sep. 7, 2020
Accepted: Nov. 14, 2020
Published Online: Jun. 21, 2021
The Author Email: Song Wei (wsong@shou.edu.cn), He Qi (qihe@shou.edu.cn)