@@ -10,33 +10,33 @@ And you can input gaussian noise to input images.
10
10
` python-tk ` (or ` python3-tk ` )
11
11
12
12
## Download dataset
13
- 1 . Download [ DF2K dataset] ( https://drive.google.com/file/d/1P9pcaGjvq3xiF22GXIq7ciZta3rjZxaY/view?usp=sharing ) .
14
- 2 . move under dataset directory i.e. vdsr_pytorch_lms /dataset/DF2K
13
+ 1 . Download [ DF2K dataset] ( https://drive.google.com/file/d/1P9pcaGjvq3xiF22GXIq7ciZta3rjZxaY/view?usp=sharing ) . Or other dataset is ok, but directory hierarchy -> ` <NAME>/train/ ` , ` <NAME>/valid/ `
14
+ 2 . move under dataset directory ( i.e. ` vdsr_pytorch /dataset/DF2K` )
15
15
16
16
## Training
17
17
```
18
- $ python main.py --dataset DF2K --cuda --gpuids 0 1 --upscale_factor 2 --crop_size 256 --batch_size 128 --test_batch_size 32
18
+ > python main.py --dataset DF2K --cuda --gpuids 0 1 --upscale_factor 2 --crop_size 256 --batch_size 128 --test_batch_size 32
19
19
```
20
20
or
21
21
```
22
- $ python3 main.py --dataset DF2K --cuda --gpuids 0 1 --upscale_factor 2 --crop_size 256 --batch_size 128 --test_batch_size 32
22
+ > python3 main.py --dataset DF2K --cuda --gpuids 0 1 --upscale_factor 2 --crop_size 256 --batch_size 128 --test_batch_size 32
23
23
```
24
24
25
25
## Test
26
26
```
27
- $ python main.py --dataset Urban100 --cuda --gpuids 0 1 --upscale_factor 2 --crop_size 256 --test_batch_size 32 --test --model model_epoch_100.pth
27
+ > python main.py --dataset Urban100 --cuda --gpuids 0 1 --upscale_factor 2 --crop_size 256 --test_batch_size 32 --test --model model_epoch_100.pth
28
28
```
29
29
or
30
30
```
31
- $ python3 main.py --dataset Urban100 --cuda --gpuids 0 1 --upscale_factor 2 --crop_size 256 --test_batch_size 32 --test --model model_epoch_100.pth
31
+ > python3 main.py --dataset Urban100 --cuda --gpuids 0 1 --upscale_factor 2 --crop_size 256 --test_batch_size 32 --test --model model_epoch_100.pth
32
32
```
33
33
34
34
## Sample usage
35
35
```
36
- $ python run.py --cuda --gpuids 0 1 --scale_factor 2 --model model_epoch_100.pth --input_image test_scale2x.jpg --output_filename test_scale2x_out.jpg
36
+ > python run.py --cuda --gpuids 0 1 --scale_factor 2 --model model_epoch_100.pth --input_image test_scale2x.jpg --output_filename test_scale2x_out.jpg
37
37
```
38
38
or
39
39
```
40
- $ python3 run.py --cuda --gpuids 0 1 --scale_factor 2 --model model_epoch_100.pth --input_image test_scale2x.jpg --output_filename test_scale2x_out.jpg
40
+ > python3 run.py --cuda --gpuids 0 1 --scale_factor 2 --model model_epoch_100.pth --input_image test_scale2x.jpg --output_filename test_scale2x_out.jpg
41
41
```
42
42
0 commit comments