Skip to content

Latest commit

 

History

History
34 lines (23 loc) · 1.14 KB

README.md

File metadata and controls

34 lines (23 loc) · 1.14 KB
Raw

Moss

Code for Paper "Memory Oriented Transfer Learning for Semi-Supervised Image Deraining"

Prerequisites

  • Python 3
  • PyTorch

Models

We provide the model trained on DDN-SIRR dataset in the following links:

Run

 CUDA_VISIBLE_DEVICES=0 python test.py --config_file='test.yaml'

Adjust the parameters in 'test.yaml' according to your own settings. The training code will be released later.

Citation

If you use our codes, please cite the following paper:

 @inproceedings{huang2021memory,
   title={Memory Oriented Transfer Learning for Semi-Supervised Image Deraining},
   author={Huang, Huaibo and Yu, Aijing and He, Ran},
   booktitle={IEEE/CVF Conference on Computer Vision and Pattern Recognition},
   pages={7732--7741},
   year={2021},
  }

The released codes are only allowed for non-commercial use.