You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+8-3Lines changed: 8 additions & 3 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -12,11 +12,13 @@ An implementation of [Enhanced Deep Residual Networks for Single Image Super-Res
12
12
13
13
## Training
14
14
In order to train, you'll have to do a few things...
15
-
- Download a dataset of images (due to my computational limitations, I've used General-100)
15
+
- Download a dataset of images (due to my computational limitations, I've used [General-100](https://drive.google.com/file/d/0B7tU5Pj1dfCMVVdJelZqV0prWnM/view?usp=drive_web))
16
16
- Place all the images from that dataset into a directory under this one
17
17
- run `python train.py --dataset data_dir` where data_dir is the directory containing your images
18
+
- In order to view stats during training (image previews, scalar for loss), simply run `tensorboard --logdir your_save_directory` where `your_save_directory`
19
+
is the directory you passed in as the save directory argument for training (`saved_models` by default)
18
20
19
-
## Training details
21
+
## Training Details
20
22
As I've mentioned before, I'm currently faced with some computational limitations, so this
21
23
caused me to do a few things differently than what is mentioned in the paper. One of the
22
24
more important changes I've made was using the General-100 dataset, because it's much smaller.
@@ -26,6 +28,9 @@ I've trained by taking the center 100x100 pixels of each image in General-100, a
26
28
I then trained an EDSR to resize the 50x50 pixel images back to 100x100. Currently, I use 80% of the
27
29
dataset as a training set and 20% as a testing set. I trained the EDSR over 1000 iterations using Adam optimizer
28
30
31
+
## Using Trained Network
32
+
In order to use trained weights you just have to run this command `python test.py`
33
+
29
34
## Results
30
35
| Original image | Shrunk image | EDSR Output |
31
36
| -------------- | ------------ | ----------- |
@@ -38,6 +43,6 @@ dataset as a training set and 20% as a testing set. I trained the EDSR over 1000
38
43
## Remarks
39
44
It seems my output images have some deconvoltion artifacts (especially around the border). I plan on finding the reason for this. It likely could be a mistake I've made.
40
45
41
-
## Future work
46
+
## Future Work
42
47
- Add MDSR implementation
43
48
- Train and post results on a larger model and dataset
0 commit comments