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

FcaNet: Frequency Channel Attention Networks

Codebase for Time-series Generative Adversarial Networks (TimeGAN)

RaceBERT -- A transformer based model to predict race and ethnicty from names

An attempt at the implementation of GLOM, Geoffrey Hinton's paper for emergent part-whole hierarchies from data

🧠 A PyTorch implementation of 'Deep CORAL: Correlation Alignment for Deep Domain Adaptation.', ECCV 2016

Official Pytorch implementation of the paper "Action-Conditioned 3D Human Motion Synthesis with Transformer VAE", ICCV 2021

Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue System

QSYM: A Practical Concolic Execution Engine Tailored for Hybrid Fuzzing

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening

Udacity Suse Cloud Native Foundations Scholarship Course Walkthrough

Generative Modelling of BRDF Textures from Flash Images [SIGGRAPH Asia, 2021]

Efficient and Accurate Arbitrary-Shaped Text Detection with Pixel Aggregation Network

A series of Python scripts to access measurements from Fluke 28X meters. Fluke IR Remote Interface required.

PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models

Align and Prompt: Video-and-Language Pre-training with Entity Prompts

This repository is to support contributions for tools for the Project CodeNet dataset hosted in DAX

Self-supervised Point Cloud Prediction Using 3D Spatio-temporal Convolutional Networks

AttentionGAN for Unpaired Image-to-Image Translation & Multi-Domain Image-to-Image Translation

CLNTM - Contrastive Learning for Neural Topic Model

Auxiliary Raw Net (ARawNet) is a ASVSpoof detection model taking both raw waveform and handcrafted features as inputs, to balance the trade-off between performance and model complexity.