Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

L-CAD: Language-based Colorization with Any-level Descriptions using Diffusion Priors

About

Language-based colorization produces plausible and visually pleasing colors under the guidance of user-friendly natural language descriptions. Previous methods implicitly assume that users provide comprehensive color descriptions for most of the objects in the image, which leads to suboptimal performance. In this paper, we propose a unified model to perform language-based colorization with any-level descriptions. We leverage the pretrained cross-modality generative model for its robust language understanding and rich color priors to handle the inherent ambiguity of any-level descriptions. We further design modules to align with input conditions to preserve local spatial structures and prevent the ghosting effect. With the proposed novel sampling strategy, our model achieves instance-aware colorization in diverse and complex scenarios. Extensive experimental results demonstrate our advantages of effectively handling any-level descriptions and outperforming both language-based and automatic colorization methods. The code and pretrained models are available at: https://github.com/changzheng123/L-CAD.

Zheng Chang, Shuchen Weng, Peixuan Zhang, Yu Li, Si Li, Boxin Shi• 2023

Related benchmarks

TaskDatasetResultRank
Image ColorizationExtended COCO-Stuff (test)
PSNR25.97
20
Image ColorizationMulti-instance (test)
PSNR25.51
20
ColorizationImageNet (test)
FID4.36
11
Visual RealismExtended COCO-Stuff (test)
Selection Rate19.72
8
Visual RealismMulti-instance (test)
Selection Rate20.28
8
Image-Description CorrespondenceExtended COCO-Stuff (test)
Selection Rate41.44
7
Image-Description CorrespondenceMulti-instance (test)
Selection Rate0.4404
7
Showing 7 of 7 rows

Other info

Code

Follow for update