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

Tensorflow Implementation of SMU: SMOOTH ACTIVATION FUNCTION FOR DEEP NETWORKS USING SMOOTHING MAXIMUM TECHNIQUE

Paddle Graph Learning (PGL) is an efficient and flexible graph learning framework based on PaddlePaddle

Multi-objective constrained optimization for energy applications via tree ensembles

Code for the submitted paper Surrogate-based cross-correlation for particle image velocimetry

Adaptable tools to make reinforcement learning and evolutionary computation algorithms.

Progressive Domain Adaptation for Object Detection

The Official Implementation of the ICCV-2021 Paper: Semantically Coherent Out-of-Distribution Detection.

NLMpy - A Python package to create neutral landscape models

pybaum provides tools to work with pytrees which is a concept burrowed from JAX.

Official PaddlePaddle implementation of Paint Transformer

Contrastive Language-Image Pretraining

Efficient Householder transformation in PyTorch

Code for Mesh Convolution Using a Learned Kernel Basis

Python implementation of "Single Image Haze Removal Using Dark Channel Prior"

FNet Implementation with TensorFlow & PyTorch

pytorch implementation of the ICCV'21 paper "MVTN: Multi-View Transformation Network for 3D Shape Recognition"

Code for Overinterpretation paper Overinterpretation reveals image classification model pathologies

Implementation of Segnet, FCN, UNet , PSPNet and other models in Keras.

Exponential Graph is Provably Efficient for Decentralized Deep Training

python library for invisible image watermark (blind image watermark)