Skip to content

tandaily/NDMamba

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NDMamba: Dual-Prior State-Space Model for Nighttime Deraining

Installation

Setting Up the Mamba Environment:

pip install causal_conv1d  
pip install mamba_ssm  

Installing the BasicSR Library:

python setup.py develop --no_cuda_ext

Datasets

GTAV-NightRain (Rerendered Version) Dataset link

Night-Rain Dataset link

RealRain-1k Dataset link

Pre-trained Models

Pre-trained model for GTAV-NightRain Dataset: model

Pre-trained model (tiny version) for GTAV-NightRain Dataset: model

Pre-trained model for Night-Rain Dataset: model

Pre-trained model for RealRain-1k Dataset: model

Visualization

Training

python basicsr/train.py -opt NightDeraining/Options/NDMamba_gtav_nightrain.yml     

Test

cd NightDeraining
python test.py

Evaluation

python eval.py

Acknowledgment

This code is built upon BasicSR, Restormer, Mamba and MambaIR. Thanks for their work.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages