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

Official Pytorch implementation of "Unbiased Classification Through Bias-Contrastive and Bias-Balanced Learning (NeurIPS 2021)

Using contrastive learning and OpenAI's CLIP to find good embeddings for images with lossy transformations

Code for the paper "Training GANs with Stronger Augmentations via Contrastive Discriminator" (ICLR 2021)

Code release for NeRF (Neural Radiance Fields)

Emotional conditioned music generation using transformer-based model.

Retinal vessel segmentation based on GT-UNet

Free course that takes you from zero to Reinforcement Learning PRO 🦸🏻‍🦸🏽

Code + pre-trained models for the paper Keeping Your Eye on the Ball Trajectory Attention in Video Transformers

Yas CRNN model training - Yet Another Genshin Impact Scanner

Patch SVDD for Image anomaly detection

Transformer Huffman coding - Complete Huffman coding through transformer

GndNet: Fast ground plane estimation and point cloud segmentation for autonomous vehicles using deep neural networks.

Whisper is a file-based time-series database format for Graphite.

This program uses trial auth token of Azure Cognitive Services to do speech synthesis for you.

Speedy Implementation of Instance-based Learning (IBL) agents in Python

Deep universal probabilistic programming with Python and PyTorch

Romanian Automatic Speech Recognition from the ROBIN project

Official PyTorch implementation of "Camera Distance-aware Top-down Approach for 3D Multi-person Pose Estimation from a Single RGB Image", ICCV 2019

Sleep staging from ECG, assisted with EEG

Code for "Adversarial attack by dropping information." (ICCV 2021)