This is the PyTorch implementation for our paper: MBA-RainGAN: A Multi-Branch Attention Generative Adversarial Network for Mixture of Rain Removal
Download the RainCityscapes++ training and validation images from Baidu Cloud. The extracting code is jpqg.
The code is coming soon.
If you find this work useful in your research, please consider cite:
@inproceedings{shen2022mba,
title={MBA-RainGAN: A Multi-Branch Attention Generative Adversarial Network for Mixture of Rain Removal},
author={Shen, Yiyang and Feng, Yidan and Wang, Weiming and Liang, Dong and Qin, Jing and Xie, Haoran and Wei, Mingqiang},
booktitle={ICASSP 2022-2022 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)},
pages={3418--3422},
year={2022},
organization={IEEE}
}
@article{shen2020mba,
title={Mba-raingan: multi-branch attention generative adversarial network for mixture of rain removal from single images},
author={Shen, Yiyang and Feng, Yidan and Deng, Sen and Liang, Dong and Qin, Jing and Xie, Haoran and Wei, Mingqiang},
journal={arXiv preprint arXiv:2005.10582},
year={2020}
}