/NeuralStyleTransfer

Implementation of Gatys' Neural Style Transfer Algorithm

Primary LanguagePython

NeuralStyleTransfer

Breaking Down Gatys' Neural Style Transfer Algorithm

Medium Article

How to run

Create the conda environment by running conda env create -f reqs.yml

Note that GPU is highly recommended to train the model. Also, remember to weight your content, style and variance losses according to desired results.

Train the model by running python nst.py --style_imge --content_image --img_dim --num_iter --style_loss_weight --content_loss_weight --variation_loss_weight --print_every --save_every by setting the flags.

Some Results

NST1

NST2

NST3

NST4