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

Relative Uncertainty Learning for Facial Expression Recognition

Code implementation from my Medium blog post: [Transformers from Scratch in PyTorch]

Models, datasets and tools for Facial keypoints detection

Semantic segmentation models, datasets and losses implemented in PyTorch.

PyTorch implementation of the paper Dynamic Data Augmentation with Gating Networks

Code to accompany the paper "Finding Bipartite Components in Hypergraphs", which is published in NeurIPS'21.

This is the official implementation of our proposed SwinMR

Controlling Hill Climb Racing with Hand Tacking

Hyperopt for solving CIFAR-100 with a convolutional neural network (CNN) built with Keras and TensorFlow, GPU backend

Out-of-Domain Human Mesh Reconstruction via Dynamic Bilevel Online Adaptation

Elucidating Robust Learning with Uncertainty-Aware Corruption Pattern Estimation

PASTRIE: A Corpus of Prepositions Annotated with Supersense Tags in Reddit International English

NALSM: Neuron-Astrocyte Liquid State Machine

The official implementation of the Interspeech 2021 paper WSRGlow: A Glow-based Waveform Generative Model for Audio Super-Resolution.

A Python parser that takes the content of a text file and then reads it into variables.

TensorFlow implementation of original paper : https://github.com/hszhao/PSPNet

Code for the paper "Generative design of breakwaters usign deep convolutional neural network as a surrogate model"

ZeroVL - The official implementation of ZeroVL

Real time Human Detection Counting

1st-in-MICCAI2020-CPM - Combined Radiology and Pathology Classification