huage001/adaattn

Public
Arbitrary Neural Style Transfer
1.6K runs

Run time and cost

Predictions run on Nvidia T4 GPU hardware. Predictions typically complete within 22 seconds.

Readme

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.

Replicate