Let's play with magic
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
conf
data
models
output
.gitignore
LICENSE
README.md
config.py
fast_evaluate.py
fast_train.py
loss.py
model.py
reader.py
util.py

README.md

Fast Style Transfer

Setup

  • Python 3+
  • Tensorflow 1.3+
  • PyYAML 3+

Transfer

  • Get trained model
  • Run python3 fast-evaluate.py --image path/to/your/image.png --model path/to/your/model/folder

Train

  • Get COCO Dataset or ImageNet Dataset
  • Put training images into data/image_train (or you can use ln -s path/to/training/images/folder data/image_train)
  • Copy conf/demo.yml to conf/your_config_file.yml, and edit it
  • Run python3 fast-train.py --conf conf/your_config_file.yml

Demo

Todo

  • Train some demo models
  • Release a single .ckpt file after training

Known Issues

  • Loss is too small to optimize

References