Skip to content

A PyTorch implementation of Adaptive Instance Normalization arbitrary style transfer.

Notifications You must be signed in to change notification settings

hvak/adaIN-style-transfer

Repository files navigation

adaIN-style-transfer

This is a pytorch implementation of Adaptive Instance Normalization (AdaIN) arbitrary style transfer, as outlined in Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization by Xun Huang and Serge Belongie.

Additionally, the optim_neural_style folder contains a pytorch implementation of A Neural Algorithm of Artistic Style by Leon A. Gatys, Alexander S. Ecker, and Matthias Bethge. This method was used as a means of comparison with AdaIN style transfer.

Style transfer comparison grid Note that our AdaIN model was trained on limited hardware and limited time. Training on a larger dataset for longer would likely produce a better result. The model that produced these AdaIN images can be found in models/best_model.pth.

More details about this project can be found in report.pdf.

Contributors

About

A PyTorch implementation of Adaptive Instance Normalization arbitrary style transfer.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages