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

🕺Full body detection and tracking

A chemical analysis of lipophilicities & molecule drawings including ML

Hard cater examples from Hopper ICLR paper

Pytorch implementation of FlowNet by Dosovitskiy et al.

Source code for the paper "Periodic Traveling Waves in an Integro-Difference Equation With Non-Monotonic Growth and Strong Allee Effect"

A simple log parser and summariser for IIS web server logs

Joint learning of images and text via maximization of mutual information

This repository contains the segmentation user interface from the OpenSurfaces project, extracted as a lightweight tool

D2LV: A Data-Driven and Local-Verification Approach for Image Copy Detection

This code is the implementation of the paper "Coherence-Based Distributed Document Representation Learning for Scientific Documents".

Official repository for "On Improving Adversarial Transferability of Vision Transformers" (2021)

Super Resolution for images using deep learning.

Speech Emotion Recognition with Fusion of Acoustic- and Linguistic-Feature-Based Decisions

Image-Stitching - Panorama composition using SIFT Features and a custom implementaion of RANSAC algorithm

Use graph-based analysis to re-classify stocks and to improve Markowitz portfolio optimization

Manage the availability of workspaces within Frappe/ ERPNext (sidebar) based on user-roles

The datasets and code of ACL 2021 paper "Aspect-Category-Opinion-Sentiment Quadruple Extraction with Implicit Aspects and Opinions".

A simple program for training and testing vit

Code for "LASR: Learning Articulated Shape Reconstruction from a Monocular Video". CVPR 2021.

Pytorch implementation of our paper accepted by NeurIPS 2021 -- Revisiting Discriminator in GAN Compression: A Generator-discriminator Cooperative Compression Scheme