Underwater single-image restoration based on modified generative adversarial net

Jindong Zhang,Dongyu Pan,Kunpeng Zhang,Jingyi Jin,Yuanzhi Ma,Mai Chen
DOI: https://doi.org/10.1007/s11760-022-02322-z
2023-04-25
Abstract:In the water medium, different light attenuation rates will cause selective absorption, resulting in poor underwater image quality. The images have shortcomings such as color distortion and blurring, which affect people's judgment. Therefore, to improve the underwater image quality, underwater single-image restoration based on modified generative adversarial net (GAN) is proposed. First, we introduce the ResNet component with small parameters into the generator to extract the deep features of the image. We adjust the order of the levels in the component and use the pre-activation method to improve the regularization ability. Secondly, the discriminator uses a local PatchGAN structure to improve image quality. To speed up the network convergence, a variety of loss functions such as GAN with Wasserstein distance and gradient penalty are introduced to jointly train the network. Finally, to improve the generalization ability of the model, the underwater images are synthesized based on the principle of the underwater dark channel prior and the underwater physical model. Experimental results show that the proposed method in this paper is better than several other underwater image restoration methods in terms of qualitative and quantitative performance.
engineering, electrical & electronic,imaging science & photographic technology
What problem does this paper attempt to address?