pixel-recursive-super-resolution
pixel-recursive-super-resolution copied to clipboard
Tensorflow implementation of pixel-recursive-super-resolution(Google Brain paper: https://arxiv.org/abs/1702.00783)
Pixel Recursive Super Resolution
TensorFlow implementation of Pixel Recursive Super Resolution. This implementation contains:
Requirements
- Python 2.7
- Skimage
- TensorFlow 1.0
Usage
First, download data celebA
$ mkdir data
$ cd data
$ ln -s $celebA_path celebA
Then, create image_list file:
$ python tools/create_img_lists.py --dataset=data/celebA --outfile=data/train.txt
To train model on gpu:
$ python tools/train.py
(or $ python tools/train.py --device_id=0)
To train model on cpu: $ python tools/train.py --use_gpu=False
Samples
Training after 30000 iteration.
Training details
cross entropy loss:
Author
nilboy / @nilboy