How to run stylize.py - Githubissues

rrmina / fast-neural-style-pytorch

Fast Neural Style Transfer implementation in PyTorch :art: :art: :art:

315 stars 77 forks source link

Closed williamzrj closed 4 years ago

williamzrj commented 4 years ago

I don't know where to put the address （--content_dir )

rrmina commented 4 years ago

Hi. There's no --content_dir argument in the repo. Did you mean to ask for the path of the content image?

williamzrj commented 4 years ago

yes

rrmina commented 4 years ago

First, please change line 9 from

STYLE_TRANSFORM_PATH = "transforms/udnie_aggressive.pth"

STYLE_TRANSFORM_PATH = "transforms/udnie.pth"

or to any style that you want and is available in the transforms folder

Then, uncomment line 135 and run the script by python stylize.py

The script will ask for the content image path while it is running. As an example, you may type images/tokyo2.jpg