Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add PReNet to PaddleGan #617

Merged
merged 16 commits into from
May 9, 2022
Next Next commit
Create prenet.md
  • Loading branch information
simonsLiang authored Apr 15, 2022
commit 00d8287e7e359d4acd98d230768339f56a79c044
49 changes: 49 additions & 0 deletions docs/en_US/tutorials/prenet.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
# PReNet

## 1 Introduction

## 2 How to use

### 2.1 Prepare dataset

The dataset(TrainH.zip) used by PReNet can be downloaded from [here](https://pan.baidu.com/s/1_vxCatOV3sOA6Vkx1l23eA?pwd=vitu),uncompress it and get two folders(RainTrainH、Rain100H).

The structure of dataset is as following::

```
├── data
├── RainTrainH
└── Rain100H
```

### 2.2 Train/Test


train model:
```
python -u tools/main.py --config-file configs/prenet.yaml
```

test model:
```
python tools/main.py --config-file configs/prenet.yaml --evaluate-only --load ${PATH_OF_WEIGHT}
```

## 3 Results

![](https://user-images.githubusercontent.com/79366697/146308440-65259d70-d056-43d4-8cf5-a82530993910.jpg)

## 4 Model Download
| 模型 | 数据集 | 下载地址 |
|---|---|---|
| PReNet | RainH.zip | [RainH.zip](https://pan.baidu.com/s/1_vxCatOV3sOA6Vkx1l23eA?pwd=vitu)




# References

- 1. [Progressive Image Deraining Networks: A Better and Simpler Baseline](https://arxiv.org/pdf/1901.09221v3.pdf)

```
```