Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
brade31919 committed Jul 17, 2017
1 parent d644600 commit fea5fb3
Show file tree
Hide file tree
Showing 4 changed files with 24 additions and 11 deletions.
7 changes: 1 addition & 6 deletions .idea/workspace.xml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

28 changes: 23 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,20 +52,26 @@ The result on BSD100, Set14, Set5 will be reported later. The code is highly ins
</table>

### Denpendency
* python2.7
* tensorflow (tested on r1.0, r1.2)
* TF slim library
* Download and extract the pre-trained model from my [google drive](https://drive.google.com/a/gapp.nthu.edu.tw/uc?id=0BxRIhBA0x8lHNDJFVjJEQnZtcmc&export=download)
* Download the VGG19 weights from the [TF-slim models](http://download.tensorflow.org/models/vgg_19_2016_08_28.tar.gz)
* The code is tested on:
* Ubuntu 14.04 LTS with CPU architecture x86_64 + Nvidia Titan X
* Ubuntu 16.04 LTS with CPU architecture x86_64 + Nvidia 1080, 1080Ti or Titan X


### Recommended
* Ubuntu 16.04 with tensorflow GPU edition

### Getting Started
Throughout the project, we denote the directory you cloned the repo as ```SRGAN-tensorflow_ROOT```<br ?>

* #### Run test using pre-trained model
```bash
# clone the repository from github
git clone https://github.com/brade31919/SRGAN-tensorflow.git
cd SRGAN-tensorflow/
cd $SRGAN-tensorflow_ROOT/

# Download the pre-trained model from the google-drive
# Go to https://drive.google.com/a/gapp.nthu.edu.tw/uc?id=0BxRIhBA0x8lHNDJFVjJEQnZtcmc&export=download
Expand All @@ -75,15 +81,15 @@ tar xvf SRGAN_pre-trained.tar
# Run the test mode
sh test_SRGAN.sh

#The result can be viewed at SRGAN-tensorflow/result/images/
#The result can be viewed at $SRGAN-tensorflow_ROOT/result/images/
```
<br />

* #### Run the training process

#### Data and checkpoint preparation
To run the training process, things will become a little complicated. Follow the steps below carefully!!<br />
Go to the project root directory. Download the vgg weight from [TF-silm model](https://github.com/tensorflow/models/tree/master/slim)<br />
Go to the project root directory. Download the vgg weight from [TF-silm model](http://download.tensorflow.org/models/vgg_19_2016_08_28.tar.gz)<br />

```bash
# make the directory to put the vgg19 pre-trained model
Expand Down Expand Up @@ -136,7 +142,11 @@ CUDA_VISIBLE_DEVICES=0 python main.py \ #Set CUDA devices correctly if you use
After ensuring the configuration. execute the script:
```bash
# Executing the script
cd $SRGAN-tensorflow_ROOT/
sh train_SRResnet.sh
```
Launch tensorboard to monitor the training process
```bash
# Launch the tensorboard
cd ./experiment_SRResnet/log/
tensorboard --logdir .
Expand Down Expand Up @@ -193,7 +203,11 @@ CUDA_VISIBLE_DEVICES=0 python main.py \
After ensuring the configuration. execute the script:
```bash
# Executing the script
cd $SRGAN-tensorflow_ROOT/
sh train_SRGAN.sh
```
Launch the tensorboard to monitor the training process
```bash
# Launch the tensorboard
cd ./experiment_SRGAN_MSE/log/
tensorboard --logdir .
Expand Down Expand Up @@ -258,7 +272,11 @@ CUDA_VISIBLE_DEVICES=0 python main.py \ #Set the place to put the c
After ensuring the configuration. execute the script:
```bash
# Executing the script
cd $SRGAN-tensorflow_ROOT/
sh train_SRGAN.sh
```
Launch tensorboard to monitor the training process
```bash
# Launch the tensorboard
cd ./experiment_SRGAN_VGG54/log/
tensorboard --logdir .
Expand Down Expand Up @@ -290,7 +308,7 @@ The training process in the tensorboard should be like this

### More result on benchmark

####Coming soon!!!
#### Coming soon!!!

### Reference
* [C. Ledig, L. Theis, F. Huszar, J. Caballero, A. Cunningham, A. Acosta, A. Aitken, A. Tejani, J. Totz, Z. Wang, W. Shi, Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network, accepted at CVPR (oral), 2017.](https://arxiv.org/pdf/1609.04802.pdf)
Expand Down
Binary file added data/test_HR/54096081_p0.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added data/test_LR/54096081_p0.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit fea5fb3

Please sign in to comment.