Skip to content

Commit 09ccb3c

Browse files
committed
update inference alone script
1 parent ab8934b commit 09ccb3c

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

README.md

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -34,14 +34,13 @@ There are some implement details with paper description, which maybe different w
3434
- [x] metrics evaluation
3535
- [x] multi-gpu support
3636
- [x] resume training / pretrained model
37+
- [x] validate alone script
3738

3839

3940

4041
## Results
4142

42-
We set the maximum reverse steps budget to 2000 now.
43-
44-
*Note:* Limited to model parameters in `Nvidia 1080Ti`, image noise and hue deviation occasionally appears in high-resolution images, resulting in low scores.
43+
*Note:* We set the maximum reverse steps budget to 2000 now. Limited to model parameters in `Nvidia 1080Ti`, **image noise** and **hue deviation** occasionally appears in high-resolution images, resulting in low scores. There are a lot room to optimization.
4544

4645
| Tasks/Metrics | SSIM(+) | PSNR(+) | FID(-) | IS(+) |
4746
| -------------------- | ----------- | -------- | ---- | ---- |
@@ -68,8 +67,6 @@ We set the maximum reverse steps budget to 2000 now.
6867

6968

7069

71-
72-
7370
## Usage
7471

7572
### Pretrained Model
@@ -87,8 +84,6 @@ This paper is based on "Denoising Diffusion Probabilistic Models", and we build
8784
"resume_state": [your pretrain model path]
8885
```
8986

90-
We have not trained the model until converged for time reason, and reduced model parameters in order to run on a single GPU, which means there are a lot room to optimization.
91-
9287
### Data Prepare
9388

9489
#### New Start
@@ -160,12 +155,18 @@ python sr.py -p train -c config/sr_sr3.json
160155
```python
161156
# Edit json to add pretrain model path and run the evaluation
162157
python sr.py -p val -c config/sr_sr3.json
158+
159+
# Quantitative evaluation alone using SSIM/PSNR metrics on given result root
160+
python eval.py -p [result root]
163161
```
164162

165-
### Evaluation Alone
163+
### Inference Alone
164+
165+
Set the HR (vanilla high resolution images), SR (images need processed) image path like step in `Own Data`. HR directory contexts can be copy from SR, and LR directory is unnecessary.
166+
166167
```python
167-
# Quantitative evaluation using SSIM/PSNR metrics on given dataset root
168-
python eval.py -p [dataset root]
168+
# run the script
169+
python infer.py -p [dataset root]
169170
```
170171

171172

0 commit comments

Comments
 (0)