Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
Dec 15, 2018
Dec 15, 2018
Dec 15, 2018
Dec 15, 2018
Dec 15, 2018
Dec 15, 2018
Dec 15, 2018

README.md

Gated Context Aggregation Network for Image Dehazing and Deraining

image

This is the implementation of our WACV 2019 paper "Gated Context Aggregation Network for Image Dehazing and Deraining" by Dongdong Chen, Mingming He, Qingnan Fan, et al.

In this paper, we propose a new end-to-end gated context aggregation network GCANet for image dehazing, in which the smoothed dilated convolution is used to avoid the gridding artifacts and a gated subnetwork is applied to fuse the features of different levels. Experiments show that GCANet can obtain much better performance than all the previous state-of-the-art image dehazing methods both qualitatively and quantitatively image

We further apply our proposed GCANet to the image deraining task, which also outperforms previous state-of-the-art image deraining methods and demonstrates its generality. image

Getting Started

This paper is implemented with Pytorch framework.

Demo

Directly put all your test images under one directory. Then run:

python test.py --task [dehaze | derain] --gpu_id [gpu_id] --indir [input directory] --outdir [output directory]

Cite

You can use our codes for research purpose only. And please cite our paper when you use our codes.

@article{chen2018gated,
  title={Gated Context Aggregation Network for Image Dehazing and Deraining},
  author={Chen, Dongdong and He, Mingming and Fan, Qingnan and Liao, Jing and Zhang, Liheng and Hou, Dongdong and Yuan, Lu and Hua, Gang},
  journal={WACV 2019},
  year={2018}
}

Contact

If you find any bugs or have any ideas of optimizing these codes, please contact me via cddlyf [at] gmail [dot] com

About

Implementation of "Gated Context Aggregation Network for Image Dehazing and Deraining"

Resources

Releases

No releases published

Packages

No packages published

Languages

You can’t perform that action at this time.