Yes, "Attention Is All You Need", for Exemplar based Colorization, ACMMM2021
Wang Yin1,
Peng Lu1,
Zhaoran Zhao1,
Xujun Peng2
1Beijing University of Posts and Telecommunications,2USC
- Ubuntu 16.04
- Python 3.6.10
- Pytorch 1.5.1
- CPU or NVIDIA GPU + CUDA 10.2 CuDNN
- Clone this repo:
git clone https://github.com/wangyins/transformer-for-image-colorization
cd transformer-for-image-colorization
pip install requriments.txt
- Download model weights from this link to get "checkpoints_acmmm2021.zip"
mkdir -p checkpoints/imagenet/
cd checkpoints/imagenet/
unzip checkpoints_acmmm2021.zip
sh test.sh
If you use this code for your research, please cite our paper.
@inproceedings{yin_mm2021,
title={Yes, "Attention Is All You Need", for Exemplar based Colorization},
author={Yin, Wang and Lu, Peng and Zhao, ZhaoRan and Peng, XuJun},
booktitle={Proceedings of the 29th ACM International Conference on Multimedia},
year={2021}
}