Unsupervised image enhancement under non-uniform illumination based on paired CNNs

Feng Lin, Huaqing Zhang, Jian Wang*, Jun Wang*

*Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

10 Citations (Scopus)

Abstract

This paper presents two CNN-based systems for unsupervised image enhancement under non-uniform illumination. The core of the systems is constituted by the difference of a pair of CNNs. Each CNN is composed of two convolutional layers of neurons with exponential activation function and logarithmic activation function. A weighted sum of the non-reference loss functions is used to train the paired CNNs. It includes an entropy enhancement function and a Bézier loss function to ensure global and local enhancement complementarily. It also includes a white balance loss function to remove color cast in raw images, and a gradient improvement loss function to compensate for the high frequency degradation. In addition, it includes an SSIM (structural similarity index) loss functions to ensure image fidelity. In addition to the basic system, CNNOD, an augmented version called CNNOD+ is developed, which features an information fusion/combination module with a power-law network for gamma correction. The experimental results on two benchmark datasets are discussed to demonstrate that the proposed systems outperform the state-of-the-art methods in terms of enhancement quality, model complexity, and convergence efficiency. © 2023 Elsevier Ltd.
Original languageEnglish
Pages (from-to)202-214
JournalNeural Networks
Volume170
Online published8 Nov 2023
DOIs
Publication statusPublished - Feb 2024

Research Keywords

  • Bézier curve
  • Image enhancement
  • Non-uniform illumination
  • Unsupervised

Fingerprint

Dive into the research topics of 'Unsupervised image enhancement under non-uniform illumination based on paired CNNs'. Together they form a unique fingerprint.

Cite this