wfs123456/GCANet

Image Derain

Gated Context Aggregation Network for Image Dehazing and Deraining

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

• Paper Link

Overview

Datasets

• Rain1200百度网盘 提取码: HZY6
• Rain1200Gooledrive
• Download datasets and put it in the folder "Derain_datasets"

Training

• python train.py

Testing

• python eval.py

Evaluation metrics

• ./scripts/PSNR.py
• ./scripts/SSIM.py

Environment

python >=3.6 
pytorch >=1.5
opencv-python >=4.0
scipy >=1.4.0
h5py >=2.10
pillow >=7.0.0
imageio >=1.18