Fast-Neural-Style-Transfer
PyTorch implementation of Fast Neural Style Transfer (official Lua implementation).
I have made some trained models available here.
Train
python3 train.py --dataset_path <path-to-dataset>
--style_image <path-to-style-image>
--epochs 1
--batch_size 4
--image_size 256
Test on Video
python3 test_on_video.py --video_path <path-to-video>
--checkpoint_model <path-to-checkpoint>
Test on Image
python3 test_on_image.py --image_path <path-to-image>
--checkpoint_model <path-to-checkpoint>