Setting Up the Mamba Environment:
pip install causal_conv1d
pip install mamba_ssm
Installing the BasicSR Library:
python setup.py develop --no_cuda_ext
GTAV-NightRain (Rerendered Version) Dataset link
Night-Rain Dataset link
RealRain-1k Dataset link
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
python basicsr/train.py -opt NightDeraining/Options/NDMamba_gtav_nightrain.yml
cd NightDeraining
python test.py
python eval.py
This code is built upon BasicSR, Restormer, Mamba and MambaIR. Thanks for their work.