Skip to content

PyTorch implementation of "Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization"

License

Notifications You must be signed in to change notification settings

edward-zhu/neural-transform

Repository files navigation

Neural Style Transfer

Train

python train.py \
    --epoch=10 \
    --batch_size=8 \
    --content_folder=./path/to/content/images \
    --style_folder=./path/to/style/images

Generate

python test.py \
    --content_folder=./path/to/content/images \
    --style_folder=./path/to/style/images \
    --model_decoder=./path/to/decoder_*.model \
    --alpha=1.0 # control the degree of stylization

About

PyTorch implementation of "Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •