Cycle-Dehaze: Enhanced CycleGAN for Single Image Dehazing
About
In this paper, we present an end-to-end network, called Cycle-Dehaze, for single image dehazing problem, which does not require pairs of hazy and corresponding ground truth images for training. That is, we train the network by feeding clean and hazy images in an unpaired manner. Moreover, the proposed approach does not rely on estimation of the atmospheric scattering model parameters. Our method enhances CycleGAN formulation by combining cycle-consistency and perceptual losses in order to improve the quality of textural information recovery and generate visually better haze-free images. Typically, deep learning models for dehazing take low resolution images as input and produce low resolution outputs. However, in the NTIRE 2018 challenge on single image dehazing, high resolution images were provided. Therefore, we apply bicubic downscaling. After obtaining low-resolution outputs from the network, we utilize the Laplacian pyramid to upscale the output images to the original resolution. We conduct experiments on NYU-Depth, I-HAZE, and O-HAZE datasets. Extensive experiments demonstrate that the proposed approach improves CycleGAN method both quantitatively and qualitatively.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Dehazing | SOTS outdoor RESIDE (test) | PSNR21.31 | 51 | |
| Image Dehazing | SOTS indoor RESIDE (test) | PSNR20.11 | 43 | |
| Single Image Desnowing | CSD 2000 (test) | PSNR20.98 | 26 | |
| Single Image Desnowing | Snow 100K 2000 (test) | PSNR26.81 | 15 | |
| Single Image Desnowing | SRRS 2000 (test) | PSNR20.21 | 15 | |
| Image Dehazing | NH-HAZE 2 NTIRE 2021 (test) | PSNR (dB)14.12 | 13 | |
| Image-to-Image Translation | GTA to Cityscapes (test) | SSIM0.71 | 10 | |
| Image Dehazing | Cholec80-Haze 1 (test) | NIQE7.96 | 10 | |
| Image Dehazing | RESIDE-OTS 31 (test) | PSNR19.95 | 10 | |
| Image Dehazing | RESIDE HSTS 31 (test) | PSNR20.95 | 10 |