Cross-domain learning for underwater image enhancement

Fei Li, Jiangbin Zheng, Yuan fang Zhang, Wenjing Jia, Qianru Wei, Xiangjian He

Research output: Journal PublicationArticlepeer-review

2 Citations (Scopus)

Abstract

The poor quality of underwater images has become a widely-known cause affecting the performance of the underwater development projects, including mineral exploitation, driving photography, and navigation for autonomous underwater vehicles. In recent years, deep learning-based techniques have achieved remarkable successes in image restoration and enhancement tasks. However, the limited availability of paired training data (underwater images and their corresponding clear images) and the requirement for vivid color correction remain challenging for underwater image enhancement, as almost all learning-based methods require paired data for training. In this study, instead of creating the time-consuming paired data, we explore the unsupervised training strategy. Specifically, we introduce a universal cross-domain GAN-based framework to generate high-quality images to address the dependence on paired training data. To ensure the vivid colorfulness, the color loss is designed to constrain the training process. Also, a feature fusion module (FFM) is proposed to increase the capacity of the whole model as well as the dual discriminator channel adopted in the architecture. Extensive quantitative and perceptual experiments show that our approach overcomes the limitation of paired data and obtains superior performance over the state-of-the-art on several underwater benchmarks in terms of both accuracy and model deployment.

Original languageEnglish
Article number116890
JournalSignal Processing: Image Communication
Volume110
DOIs
Publication statusPublished - Jan 2023
Externally publishedYes

Keywords

  • GAN
  • Loss function
  • Underwater image enhancement
  • Unsupervised learning

ASJC Scopus subject areas

  • Software
  • Signal Processing
  • Computer Vision and Pattern Recognition
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Cross-domain learning for underwater image enhancement'. Together they form a unique fingerprint.

Cite this