This is the Pytorch implementation of Universal Style Transfer via Feature Transforms.
Official Torch implementation can be found here and Tensorflow implementation can be found here.
- Pytorch
- torchvision
- Pretrained encoder and decoder models for image reconstruction only (download and uncompress them under models/)
- CUDA + CuDNN
Simply put style images in images/style/style_image
python WCT-capture.py
Many thanks to the author Yijun Li for his kind help.
Li Y, Fang C, Yang J, et al. Universal Style Transfer via Feature Transforms[J]. arXiv preprint arXiv:1705.08086, 2017.