Skip to content

crypto-code/Style-Transfer

Repository files navigation

Style-Transfer

Use Neural Networks to Paint Images in Famous Styles

For more info: Research Paper

Requirements:

Installation:

  • Download and extract entire repositry
  • Open cmd and run the following to install required modules
pip install -r requirements.txt 
  • Download Pre-trained VGG network and put it in the main file, or specify its location using the --network option.

  • Once requirements are installed download a painting of any style you need, eg- Scream (Already provided) and the original image to be painted (Golden Gate Bridge).

Running

python style_transfer.py --content <content file> --styles <style file> --output <output file>

Run python style_transfer.py --help to see a list of all options.

Use --checkpoint-output and --checkpoint-iterations to change checkpoint save location and iteration (default: 'output/output_{:05}.jpg', 10).

Use --iterations to change the number of iterations (default: 500).

Examples:

With Golden Gate Bridge and the Scream

With Emma Watson and Mona Lisa

G00D LUCK

For doubts email me at: atinsaki@gmail.com