Single Underwater Image Restoration via Unsupervised Generative Adversarial Network and Contrastive Learning

Yi Hung Sung*, Li Wei Kang*, Chia Hung Yeh

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Underwater image restoration has gained more and more attention recently due to its several applications in marine environmental surveillance-related tasks. In this paper, a novel unsupervised GAN (generative adversarial network)-based deep learning framework for single underwater image restoration is proposed. Without needing paired training images, we introduce contrastive learning with feature and style reconstruction loss functions in our unsupervised GAN-based structure to learn an image generator for translating underwater images to the corresponding in-air images. Extensive experiments have shown that the proposed method outperforms (or is comparable with) the state-of-the-art deep learning-based methods relying on paired/unpaired training data quantitatively and qualitatively.

Original languageEnglish
Title of host publication2023 International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2023 - Proceedings
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages853-854
Number of pages2
ISBN (Electronic)9798350324174
DOIs
Publication statusPublished - 2023
Event2023 International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2023 - Pingtung, Taiwan
Duration: 2023 Jul 172023 Jul 19

Publication series

Name2023 International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2023 - Proceedings

Conference

Conference2023 International Conference on Consumer Electronics - Taiwan, ICCE-Taiwan 2023
Country/TerritoryTaiwan
CityPingtung
Period2023/07/172023/07/19

Keywords

  • contrastive learning
  • deep learning
  • generative adversarial networks
  • single underwater image restoration
  • unsupervised learning

ASJC Scopus subject areas

  • Artificial Intelligence
  • Human-Computer Interaction
  • Information Systems
  • Information Systems and Management
  • Electrical and Electronic Engineering
  • Media Technology
  • Instrumentation

Fingerprint

Dive into the research topics of 'Single Underwater Image Restoration via Unsupervised Generative Adversarial Network and Contrastive Learning'. Together they form a unique fingerprint.

Cite this