FastNeuralStyle by Pytorch

Fast Neural Style for Image Style Transform by Pytorch

This is famous Fast Neural Style of Paper Perceptual Losses for Real-Time Style Transfer and Super-Resolution of Feifei Li.

Result

Style Image | Origin Image | Generated Image |:---:|:---:|:---:| | ||

How to Run

Train and Test

Training DataSet

I Strongly Recommend you to download coco80k 2014 dataset from coco80k 2014 (This is also used by the original paper)

You can use your own huge image dataset as well

Configure

put dataset under ./dataset, the folder will looks like

dataset 
 --train2014
 --put_coco_train_here

Run

train 
python train.py -h

test
python go.py -h

Pretrained Model

models are saved in ./model folder by default.

I also release two pretrained model .

wave.model

You can download and put it under ./model folder