Selective Wavelet Attention Learning for Single Image Deraining

Last update: Jun 17, 2022

Overview

SWAL

Code for Paper "Selective Wavelet Attention Learning for Single Image Deraining"

Prerequisites

Python 3
PyTorch

Models

We provide the models trained on DDN, DID, Rain100H, Rain100L, and AGAN datasets in the following links:

Download them into the model folder before testing.

Dataset

Download the rain datasets.
Arrange the images and generate a list file, just like the rain12 set in the data folder.

You can also modify the data_loader code in your manner.

Run

Train SWAL on a single GPU:

 CUDA_VISIBLE_DEVICES=0 python main.py --ngf=16 --ndf=64  --output_height=320  --trainroot=YOURPATH --trainfiles='YOUR_FILELIST'  --save_iter=1 --batchSize=8 --nrow=8 --lr_d=1e-4 --lr_g=1e-4  --cuda  --nEpochs=500

Train SWAL on multiple GPUs:

 CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py --ngf=16 --ndf=64  --output_height=320  --trainroot=YOURPATH --trainfiles='YOUR_FILELIST'  --save_iter=1 --batchSize=32 --nrow=8 --lr_d=1e-4 --lr_g=1e-4  --cuda  --nEpochs=500

Test SWAL:

 CUDA_VISIBLE_DEVICES=0 python test.py --ngf=16  --outf='test' --testroot='data/rain12_test' --testfiles='data/rain12_test.list' --pretrained='model/rain100l_best.pth'  --cuda

Adjust the parameters according to your own settings.

Citation

If you use our codes, please cite the following paper:

 @article{huang2021selective,
   title={Selective Wavelet Attention Learning for Single Image Deraining},
   author={Huang, Huaibo and Yu, Aijing and Chai, Zhenhua and He, Ran and Tan, Tieniu},
   journal={International Journal of Computer Vision},
   volume={129},
   number={4},
   pages={1282--1300},
   year={2021},
  }

The released codes are only allowed for non-commercial use.

Selective Wavelet Attention Learning for Single Image Deraining

Related tags

Overview

SWAL

Prerequisites

Models

Dataset

Run

Citation

Owner

Bobo

The Pytorch code of "Joint Distribution Matters: Deep Brownian Distance Covariance for Few-Shot Classification", CVPR 2022 (Oral).

Torch-mutable-modules - Use in-place and assignment operations on PyTorch module parameters with support for autograd

Easily Process a Batch of Cox Models

MAUS: A Dataset for Mental Workload Assessment Using Wearable Sensor - Baseline system

SynNet - synthetic tree generation using neural networks

This is the official implementation of "One Question Answering Model for Many Languages with Cross-lingual Dense Passage Retrieval".

FluidNet re-written with ATen tensor lib

TorchX: A PyTorch Extension Library for More Efficient Deep Learning

Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR 2018 (spotlight)

Simply enable or disable your Nvidia dGPU

Image Captioning using CNN and Transformers

Keras implementations of Generative Adversarial Networks.

Code for Talking Face Generation by Adversarially Disentangled Audio-Visual Representation (AAAI 2019)

USAD - UnSupervised Anomaly Detection on multivariate time series

RL and distillation in CARLA using a factorized world model

Software that can generate photos from paintings, turn horses into zebras, perform style transfer, and more.

PyStan, a Python interface to Stan, a platform for statistical modeling. Documentation: https://pystan.readthedocs.io

Fuzzification helps developers protect the released, binary-only software from attackers who are capable of applying state-of-the-art fuzzing techniques

Improving Factual Consistency of Abstractive Text Summarization

Code to reproduce the results for Compositional Attention