Skip to content

A Common Implementation of StyleTransfer in Pytorch

Notifications You must be signed in to change notification settings

rzamarefat/StyleTransfer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

StyleTransfer

A Common Implementation of StyleTransfer in Pytorch

Sample 1

StyleTransfer

Sample 2

StyleTransfer2 (1)

How to use

1- Make an conda env 2- Install Pyotorch and torchvision 3- Go ahead and install the following packages

pip install tqdm opencv-python 

After installing all the dependencies run the following command:

python train.py --path-to-original-image path/to/original/image --path-to-style-image --path/to/style/image --path-to-save-output-image --path/to/save/output/image --num-steps 10000

Please refer to train.py module to see the other options.

About

A Common Implementation of StyleTransfer in Pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages