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 .
You can download and put it under ./model folder