Home

Awesome

AAAI-2020-FBL-SS

This website shares the codes of the "Towards Scale-Free Rain Streak Removal via Self-Supervised Fractal Band Learning", AAAI-2020.

Prerequisites:

  1. Linux
  2. Python 3.5
  3. NVIDIA GPU + CUDA CuDNN (CUDA>8.0)

Installation:

  1. Clone this repo
  2. Install PyTorch and dependencies from http://pytorch.org

Note: the code is suitable for PyTorch 0.4.0)

Usage:

  1. For the usage of FBL, you can access to ./FBL/src/train_DID_MDN.sh, ./FBL/src/train_DID_MDN.sh, ./FBL/src/train_DetailNet.sh, ./FBL/src/test_DID_MDN.sh, ./FBL/src/test_DetailNet.sh.

  2. For the usage of FBL-SS, you can access to ./FBL-ss/src/train_free.sh and ./FBL-ss/src/test.sh.

Contact

If you have questions, you can contact yangwenhan@pku.edu.cn.