Underwater image enhancement model based on generative adversarial network with key feature transfer
DOI:
CSTR:
Author:
Affiliation:

1.College of Mathematics and Computer Science, Guangdong Ocean University, Zhanjiang 524088, China; 2.College of Electronics and Information Engineering, Guangdong Ocean University, Zhanjiang 524088, China; 3. Research Center of Guangdong Smart Oceans Sensor Networks and Equipment Engineering, Zhanjiang 524088, China

Clc Number:

TP391;TN919.8

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    To address the challenges of color cast, blurriness, and detail loss in underwater images caused by light absorption and scattering, this study proposes a KFT-GAN for underwater image enhancement. The paper introduces a KFT module that facilitates the efficient transmission of critical features such as color, edges, and texture within the network. By integrating depthwise separable convolution, a lightweight network model is constructed, reducing the parameter count by 60.6% compared to models utilizing conventional convolution, thereby enhancing the model’s learning efficiency. This module enables the generator’s encoding stage to extract essential key features from the input image and transmit these features to the decoding stage through downsampling and skip connections, improving the quality of the reconstructed image. Additionally, this study proposes a hybrid loss function based on perceptual loss principles, emphasizing multiple key attributes of the image to achieve superior visual quality. The proposed model demonstrates excellent performance on both the EUVP and UIEB datasets, achieving PSNR values of 21.384 2 and 18.025 6, and SSIM values of 0.741 3 and 0.688 9, respectively. Qualitative and quantitative comparisons with traditional algorithms and deep learning methods validate the effectiveness and superiority of the proposed model.

    Reference
    Related
    Cited by
Get Citation
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: April 30,2026
  • Published:
Article QR Code