Journal of Shanghai Jiao Tong University ›› 2022, Vol. 56 ›› Issue (2): 134-142.doi: 10.16183/j.cnki.jsjtu.2021.075

Previous Articles     Next Articles

Underwater Image Enhancement Based on Generative Adversarial Networks

LI Yu, YANG Daoyong, LIU Lingya, WANG Yiyin()   

  1. School of Electronic Information and Electrical Engineering, Shanghai Jiao Tong University, Shanghai 200240, China
  • Received:2020-03-11 Online:2022-02-28 Published:2022-03-03
  • Contact: WANG Yiyin


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.

Key words: underwater image enhancement, generative adversarial networks, residual structure, unsupervised learning

CLC Number: