Skip to content

Commit edf3c3f

Browse files
committed
Update README
1 parent ca99e58 commit edf3c3f

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

README.md

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,13 @@ An implementation of [Enhanced Deep Residual Networks for Single Image Super-Res
1212

1313
## Training
1414
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))
1616
- Place all the images from that dataset into a directory under this one
1717
- 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)
1820

19-
## Training details
21+
## Training Details
2022
As I've mentioned before, I'm currently faced with some computational limitations, so this
2123
caused me to do a few things differently than what is mentioned in the paper. One of the
2224
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
2628
I then trained an EDSR to resize the 50x50 pixel images back to 100x100. Currently, I use 80% of the
2729
dataset as a training set and 20% as a testing set. I trained the EDSR over 1000 iterations using Adam optimizer
2830

31+
## Using Trained Network
32+
In order to use trained weights you just have to run this command `python test.py`
33+
2934
## Results
3035
| Original image | Shrunk image | EDSR Output |
3136
| -------------- | ------------ | ----------- |
@@ -38,6 +43,6 @@ dataset as a training set and 20% as a testing set. I trained the EDSR over 1000
3843
## Remarks
3944
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.
4045

41-
## Future work
46+
## Future Work
4247
- Add MDSR implementation
4348
- Train and post results on a larger model and dataset

0 commit comments

Comments
 (0)