Transformer for Image Colorization

This is an implemention for Yes, “Attention Is All You Need”, for Exemplar based Colorization, and the current software works well with PyTorch 1.5.1.

Table of Contents

Prerequisites

  • Ubuntu 16.04
  • Python 3.6.10
  • CPU or NVIDIA GPU + CUDA 10.2 CuDNN

Getting Started

Installation

  • Clone this repo:

git clone https://github.com/wangyin-cv/transformer-for-image-colorization
cd transformer-for-image-colorization
pip install requriments.txt
  • Download model weights from [Google Drive] to obtain “checkpoints_acmmm2021.zip”

mkdir -p checkpoints/imagenet/
cd checkpoints/imagenet/
unzip checkpoints_acmmm2021.zip

Testing

sh test.sh

Citation

If you use this code for your research, please cite our papers.

@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}
}

GitHub

View Github