Skip to content

Latest commit

 

History

History
16 lines (13 loc) · 661 Bytes

README.md

File metadata and controls

16 lines (13 loc) · 661 Bytes

Neural Style Transfer

The repositiory implements the method for Neural Style Transfer based on the original paper

Dependencies

This is assuming you have pytorch already setup

  pip install tqdm
  pip install numpy

Running Tests

Download or clone the github repository and the above listed dependencies. To run, execute the folloing command

  python main.py --style_img_pth YOUR_STYLE_IMAGE_PATH --content_img_pth YOUR_CONTENT_IMAGE_PATH --total_steps TRAINING_STEPS --learning_rate YOUR_LR --image_size OUTPUT_IMG_SIZE