EnsembleNet

Installation

Framework

1. Python 3.7
2. Pytorch1.0 (with ubuntu 16)
3. Torchvision

Python Dependencies

1. skimage
2. numpy
3. visdom : pip install visdom

Train

python3.7 Derain.py 

Test

python3.7 Derain.py 

Citation:

Ensemble single image deraining network via progressive structural boosting constraints

  @article{peng2021ensemble,
    title={Ensemble single image deraining network via progressive structural boosting constraints},
    author={Peng, Long and Jiang, Aiwen and Wei, Haoran and Liu, Bo and Wang, Mingwen},
    journal={Signal Processing: Image Communication},
    pages={116460},
    year={2021},
    publisher={Elsevier}
    }
    
    IF it helps you , please quotes us , thanks a lot