Underwater Image Enhancement Based on Generative Adversarial Networks

Abstract

This paper proposes an underwater image correction and enhancement algorithm based on generative adversarial networks. In this algorithm, the multi-scale kernel is applied to the improved residual module to construct a generator, which realizes the extraction and fusion of multiple receptive fields feature information. The discriminator design considers the relationship between global information and local details, and establishes a global-region dual discriminator structure, which can ensure the consistency of overall style and edge texture. An unsupervised loss function based on human visual sensory system is proposed. Reference image constraints are not required, and the confrontation loss and the content loss are jointly optimized to obtain better color and structure performance. Experimental evaluations on multiple data sets show that the proposed algorithm can better correct color deviation and contrast, protect details from loss, and is superior to typical algorithms in subjective and objective indexes.

Keywords

• underwater image enhancement
• generative adversarial networks
• residual structure
• unsupervised learning