yunguo224/UG2_Deraining

[CVPRW 2023] Official solution of Team HUST_VIE for UG2+ Challenge Track 3 GT-RAIN Challenge

UG2_Deraining

[CVPRW 2023] Official solution of Team HUST_VIE for UG2+ Challenge Track 3 GT-RAIN Challenge

Package dependencies

You can install by:

pip install -r requirements.txt

Pretrained model

The pre-trained model is provided in Releases. Download and place in the root directory.

Evaluation

To evaluate, you can run:

python test_derain_Uformer.py

The test file should contain input/groundtruth files.