Underwater Image Enhancement Using Image Formation Model and GAN

Yang Zhang,Hongshuai Pang,Wenping Cao,Kewei Cai,Tao Zhang,Zhiqiang Wang
DOI: https://doi.org/10.1007/978-981-19-3171-0_42
2022-01-01
Abstract:Underwater vision has played an important role in robotic operations, exploration of seabed resources, sea life monitoring and underwater fishing. However, raw underwater images, which suffer from color distortion and poor image quality, make underwater vision more challengeable than open air vision. To address the mentioned problems, a novel model for underwater image color correction is proposed in this paper. Firstly, a synthetic underwater image generating network is proposed to address the lack of effective underwater training data problem. The model combines a background color prior with an underwater imaging model to generate synthetic underwater images by using in-air images data. By the benefit of the color prior and imaging model, the generating model can be with fewer number of parameters and more effective. Meanwhile, a simple yet effective model is proposed to train on the in-air images and corresponding rendered synthetic underwater images for color correction. Different from other image restoration models with multiple substructure or complicated construction, the proposed method has only few convolution layers. Finally, the enhanced results demonstrate the superiority of the proposed method, which performs favorably against the existing stage-of-the-art methods in both effectiveness and model size aspects.
What problem does this paper attempt to address?