Usage

Requirements

• Install skimage via pip: sudo pip install scikit-image
• Install CUDA and Cudnn
• Install Mxnet, please set "USE_NVRTC = 1" in config.mk before compiling
• Download pretrained VGG model and save it to the root of this repository.
• Download MSCOCO dataset if you want to train models.

Use Trained Models

To use trained models to generate images, first load the model and specify the output shape.

python generate.py

Train New Model

First, edit train_gram.py, modify VGGPATH, CONTENTPATH and STYLEPATH

python train_gram.py