Skip to content

Pytorch implementation of Neural Network of Artistic Style

License

Notifications You must be signed in to change notification settings

djin31/StyleTransfer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

StyleTransfer

Pytorch implementation of style transfer technique proposed by Gatys et al, 2016.

surfer X waves

Requirements

  • torch
  • torchvision
  • PIL

Usage

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.

Examples

Some cool examples generated by the model:

IIT Delhi campus stylized as Van Gogh's The Starry Night. iitd X starry_night

City of Paris regenerated with inspiration from a painting of the city paris X paris painting

Professor from Money Heist inspired by Dali. professor X dali

About

Pytorch implementation of Neural Network of Artistic Style

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published