The code of the paper "Wavelet Frequency Division Self-attention Transformer Image Deraining Network"
论文《小波分频自注意力Transformer图像去雨网络》的代码
python = 3.7
pytorch = 1.9
opencv-python = 4.5
numpy = 1.21
Run shell scripts to train the models
python train.py
Run shell scripts to eval the models
python eval.py
Run shell scripts to show rain removal effect"
python show.py
Rain200H: Link
Rain200L: Link
Rain1200: Link
Rain12: Link
SPA-Data: Link
Rain200H_Model: Link
VOCRain250: Link
This code is based on the RESCAN. Thank them for their sharing.
If you have any questions, please contact fangsiyanfsy@163.com