Use Neural Networks to Paint Images in Famous Styles
For more info: Research Paper
- Python 3.6.2 (https://www.python.org/downloads/release/python-362/)
- Numpy (https://pypi.org/project/numpy/)
- Tensorflow (https://pypi.org/project/tensorflow/)
- Keras (https://pypi.org/project/Keras/)
- 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).
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).
With Golden Gate Bridge and the Scream
With Emma Watson and Mona Lisa
For doubts email me at: atinsaki@gmail.com