project-athenians

Style Transfer

Implementation of paper Universal Style Transfer via Feature Transforms, Li, Y., Fang, C., Yang, J., Wang, Z., Lu, X., & Yang, M. H. (2017)

Repository structure

├── docs
├── results
├── src
	├── images
	├── models
	├── main.py
	├──other python files
├── environment.yml

Instructions to run the code

• clone repository
• install requirements

conda env create -f environment.yml

• download content and style images and name the content image same as corresponding style image.
• cd src
• run main.py

python main.py --contentPath <path to content images> --stylePath <path to style images> --outf <path for saving result> --cuda

example:

python main.py --cuda

python main.py --cuda --stylePath images/style --outf images/samples --cuda

To add style to image with user control of mask, do

python run_with_masks.py --masksContentPath <path to content image> --masksStylePath <path to style image directory> --masksResultPath <path to result generated with mask> --cuda