Pytorch implementation of style transfer technique proposed by Gatys et al, 2016.
- torch
- torchvision
- PIL
python style_transfer.py --content-path [path to content image] --style-path [path to style image]
Run python style_transfer.py -h for list of optional arguments.
Example usage
python style_transfer.py --content-path ./examples/obama.jpg --style-path ./examples/van-gogh.jpg --output "./examples/obama X van-gogh/" --use-gpu --num-steps 200
Check out Jupyter notebook for description of the style transfer technique.
Some cool examples generated by the model:
IIT Delhi campus stylized as Van Gogh's The Starry Night.
City of Paris regenerated with inspiration from a painting of the city
Professor from Money Heist inspired by Dali.
