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

Auto-Lama combines object detection and image inpainting to automate object removals

Minimal deep learning library written from scratch in Python, using NumPy/CuPy.

An open source machine learning library for performing regression tasks using RVM technique.

Disentangled Cycle Consistency for Highly-realistic Virtual Try-On, CVPR 2021

Visualizing lattice vibration information from phonon dispersion to atoms (For GPUMD)

Official codebase for Legged Robots that Keep on Learning: Fine-Tuning Locomotion Policies in the Real World

Code release for Universal Domain Adaptation(CVPR 2019)

Code for the TIP 2021 Paper "Salient Object Detection with Purificatory Mechanism and Structural Similarity Loss"

Pytorch code for "DPFM: Deep Partial Functional Maps" - 3DV 2021 (Oral)

Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers

UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset

This is a simple face recognition mini project that was completed by a team of 3 members in 1 week's time

Using image super resolution models with vapoursynth and speeding them up with TensorRT

Trying to understand alias-free-gan.

Tools for investing in Python

Repo for "Physion: Evaluating Physical Prediction from Vision in Humans and Machines" submission to NeurIPS 2021 (Datasets & Benchmarks track)

A Unified Framework and Analysis for Structured Knowledge Grounding

The official pytorch implementation of our paper "Is Space-Time Attention All You Need for Video Understanding?"

maximal update parametrization (µP)

A universal framework for learning timestamp-level representations of time series