Style_Transformer

Paper

Perceptual Losses for Real-Time Style Transfer and Super-Resolution: https://arxiv.org/pdf/1603.08155.pdf

Train model

you need to create a dir of save image data for the trained model and create a dir of saving models.

bash train.sh

Test model

bash test.sh

data

-- datas

---train2014

数据加载链接:https://cocodataset.org/#download;

Reference

https://github.com/pytorch/examples/tree/78acb79062189bd937f17edf7c97571e6ec59083/fast_neural_style