zanvari / ecdn

Enhanced CycleGAN Dehazing Network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Enhanced CycleGAN Dehazing Network

We focus on unpaired single image dehazing and reduce the image dehazing problem to an unpaired image-to-image translation and propose an Enhanced CycleGAN Dehazing Network (ECDN). We enhance CycleGAN from different angles for the dehazing purpose. We employ a global-local discriminator structure to deal with spatially varying haze. We define self-regularized color loss and utilize it along with perceptual loss to generate more realistic and visually pleasing images. We use an encoder-decoder architecture with residual blocks in the generator with skip connections so that the network better preserves the details. Through an ablation study, we demonstrate the effectiveness of different modules in the performance of the proposed network. Our extensive experiments over two benchmark datasets show that our network outperforms previous work in terms of PSNR and SSIM. Paper is here.

ECDN Architecture

Overview

alt text

Generator Architecture

alt text

Results

alt text

Publication

If you find this work useful for you, please cite:

@InProceedings{anvari_visapp2021,
    author    = {Anvari, Zahra and Athitsos, Vassilis},
    title     = {Enhanced CycleGAN Dehazing Network},
    booktitle = {Proceedings of the 16th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISAPP)},
    month     = {February},
    year      = {2021},
    volume    = {4},
    pages     = {193-202}
}

About

Enhanced CycleGAN Dehazing Network


Languages

Language:Python 100.0%