iArunava/Style-Transfer-with-PyTorch

Implementing the Style Transfer Paper by Gatys in PyTorch. Using pretrained models to transfer arbitrary style onto arbitrary images.

Style-Transfer-with-PyTorch

Implementing the Style Transfer Paper by Gatys in PyTorch.
Link to paper: https://www.cv-foundation.org/openaccess/content_cvpr_2016/papers/Gatys_Image_Style_Transfer_CVPR_2016_paper.pdf Using pretrained models to transfer arbitrary style onto arbitrary images. The project allows one to start training and see save results from just the command line with one simple command!!

How to use

1. Clone the repo

git clone https://github.com/iArunava/Style-Transfer-with-PyTorch.git

2. cd to the repo

cd Style-Transfer-with-PyTorch

3. Start Style Transfer

python3 style_transfer.py -c='/path/to/content/image' -s='/path/to/style/image'

4. For more involved use --help

python3 style_transfer.py -c='/path/to/content/image' -s='/path/to/style/image' -lr=0.001 -e 1000

Have fun!

Examples

Content Image: Photo by Olivier Guillard on Unsplash

Style Image: Photo by Ashkan Forouzani on Unsplash

Stylized Image: Photo by an Algorithm ;)

License

The code in this repository is distributed under GPL 3.0
Feel free to fork the repository and perform your own style transfers!!