This is an exercise in the Udacity course, Intro to Deep Learning with PyTorch. This repo is a simple implementation of performing image style transformation in PyTorch. Specifically, I take a pre-trained VGG-19 model as a feature extractor and use it to extract content and style features from a content and style image, respectively. The transferred image is crafted in a way minimizing the content and style loss.
This script has been tested on anaconda3 and PyTorch 0.4.0.
Transferred Image Generation:
python style_transfer.py --content_img_path=<path to a content image>
--style_img_path=<path to a style image>
--output_img_dir_path=<path to an output image directory>
Content Image: space needle, Style Image: kahlo
