Enhanced CycleGAN Network with Adaptive Dark Channel Prior for Unpaired Single-Image Dehazing

Entropy (Basel). 2023 May 26;25(6):856. doi: 10.3390/e25060856.

Abstract

Unpaired single-image dehazing has become a challenging research hotspot due to its wide application in modern transportation, remote sensing, and intelligent surveillance, among other applications. Recently, CycleGAN-based approaches have been popularly adopted in single-image dehazing as the foundations of unpaired unsupervised training. However, there are still deficiencies with these approaches, such as obvious artificial recovery traces and the distortion of image processing results. This paper proposes a novel enhanced CycleGAN network with an adaptive dark channel prior for unpaired single-image dehazing. First, a Wave-Vit semantic segmentation model is utilized to achieve the adaption of the dark channel prior (DCP) to accurately recover the transmittance and atmospheric light. Then, the scattering coefficient derived from both physical calculations and random sampling means is utilized to optimize the rehazing process. Bridged by the atmospheric scattering model, the dehazing/rehazing cycle branches are successfully combined to form an enhanced CycleGAN framework. Finally, experiments are conducted on reference/no-reference datasets. The proposed model achieved an SSIM of 94.9% and a PSNR of 26.95 on the SOTS-outdoor dataset and obtained an SSIM of 84.71% and a PSNR of 22.72 on the O-HAZE dataset. The proposed model significantly outperforms typical existing algorithms in both objective quantitative evaluation and subjective visual effect.

Keywords: atmospheric scattering model; cycle generative adversarial network; dark channel prior; dehaze.

Grants and funding

This work is supported by the Venture and Innovation Support Program for Chongqing Overseas Returnees (Grant No. cx2019133) and the Chongqing Research Project of the Foal Eagle Program (Grant No. CY220231).