🚀 Want to run this model with an API? Get started

huage001/adaattn

Public
Arbitrary Neural Style Transfer
3K runs

Run time and cost

Predictions run on Nvidia T4 GPU hardware. Predictions typically complete within 3 seconds. The predict time for this model varies significantly based on the inputs.

AdaAttN: Revisit Attention Mechanism in Arbitrary Neural Style Transfer

Overview

This repository contains the officially unofficial PyTorch re-implementation of paper:

AdaAttN: Revisit Attention Mechanism in Arbitrary Neural Style Transfer,

Songhua Liu, Tianwei Lin, Dongliang He, Fu Li, Meiling Wang, Xin Li, Zhengxing Sun, Qian Li, Errui Ding

ICCV 2021

Citation

If you find ideas or codes useful for your research, please cite:

```
@inproceedings{liu2021adaattn,
  title={AdaAttN: Revisit Attention Mechanism in Arbitrary Neural Style Transfer},
  author={Liu, Songhua and Lin, Tianwei and He, Dongliang and Li, Fu and Wang, Meiling and Li, Xin and Sun, Zhengxing and Li, Qian and Ding, Errui},
  booktitle={Proceedings of the IEEE International Conference on Computer Vision},
  year={2021}
}
```

Acknowledgments

This implementation is developed based on the code framework of pytorch-CycleGAN-and-pix2pix by Junyan Zhu et al.