This is the author's official PyTorch L-CoDe implementation. In this paper, we propose L-CoDe, a Language-based Colorization network using color-object Decoupled conditions. A predictor for object-color corresponding matrix (OCCM) and a novel attention transfer module (ATM) are introduced to solve the color-object coupling problem. To deal with color-object mismatch that results in incorrect color-object correspondence, we adopt a soft-gated injection module (SIM).
- Python 3.6
- PyTorch 1.0
- NVIDIA GPU + CUDA cuDNN
Clone this repo
Install PyTorch and dependencies from https://pytorch.org
Install other python requirements
We process the MSCOCO dataset for evaluation. Specifically, we keep the images whose captions contain adjectives and annotate the correspondence between adjectives and nouns in the caption to produce the ground-truth object-color corresponding matrix (OCCM). Metadata is in here.
Download the coco2017 images and copy them under IMG_DIR.
Setting the MODEL_DIR as the storage directory for generated experimental results.
These directory parameters could be found in cfg/coco_train.yml and cfg/coco_test.yml.
python train.py --gm --o2c --train
python test.py --gm --o2c
Licensed under a Creative Commons Attribution-NonCommercial 4.0 International.
Except where otherwise noted, this content is published under a CC BY-NC license, which means that you can copy, remix, transform and build upon the content as long as you do not use the material for commercial purposes and give appropriate credit and provide a link to the license.
If you use this code for your research, please cite our papers L-CoDe: Language-based Colorization Using Color-object Decoupled Conditions
@InProceedings{Weng_2022_AAAI,
author = {Weng, Shuchen and Wu, Hao and Chang, Zheng and Tang, Jiajun and Li ,Si and Shi, Boxin},
title = {L-CoDe: Language-based Colorization Using Color-object Decoupled Conditions},
booktitle = {AAAI},
month = {June},
year = {2022}
}