shaun95/adain-pytorch

PyTorch implementation of "Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization" by Xun Huang, Serge Belongie.

AdaIN-pytorch

PyTorch implementation of "Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization" by Xun Huang, Serge Belongie. arxiv.

0. References

• irasin/Pytorch_AdaIN

1. Set up and train model

1.1 install requirements

pip install -r requitements.txt

1.2 get the datasets and unzip them to desired location

• COCO: https://cocodataset.org/#download
• WikiArt: https://www.kaggle.com/c/painter-by-numbers/data

1.3 Train model

• to start training: python main.py train path/to/coco path/to/wikiart [OPTIONS]

• to change training parameters and options: python main.py train --help

2. Style Transfer using pre-trained model

• to run inference: python main.py infer [OPTIONS]
• info on supported options: python main.py infer --help

3. Some Results

• All images were resized to 1024x1024 for inference. The 1024x1024 outputs are interpolated to the original size using bilinear interpolation.

Content Img	Style Img	Output