MULTIMODAL SEMANTIC-AWARE AUTOMATIC COLORIZATION WITH DIFFUSION PRIOR

Shanghai Jiao Tong University
*Corresponding Author

Showcases produced by our pipeline.

Abstract

Colorizing grayscale images offers an engaging visual experience. Existing automatic colorization methods often fail to generate satisfactory results due to incorrect semantic colors and unsaturated colors. In this work, we propose an automatic colorization pipeline to overcome these challenges. We leverage the extraordinary generative ability of the diffusion prior to synthesize color with plausible semantics. To overcome the artifacts introduced by the diffusion prior, we apply the luminance conditional guidance. Moreover, we adopt multimodal high-level semantic priors to help the model understand the image content and deliver saturated colors. Besides, a luminance-aware decoder is designed to restore details and enhance overall visual quality. The proposed pipeline synthesizes saturated colors while maintaining plausible semantics. Experiments indicate that our proposed method considers both diversity and fidelity, surpassing previous methods in terms of perceptual realism and gain most human preference.

Framework

MY ALT TEXT

Overview of the proposed automatic colorization pipeline. It combines a semantic prior generator (blue box), a high-level semantic guided diffusion model(yellow box), and a luminance-aware decoder (orange box).

Evaluations

Qualitative

Quantitative

Ablation

Discussion on parameter i.

BibTeX

@misc{wang2024multimodal,
        title={Multimodal Semantic-Aware Automatic Colorization with Diffusion Prior}, 
        author={Han Wang and Xinning Chai and Yiwen Wang and Yuhong Zhang and Rong Xie and Li Song},
        year={2024},
        eprint={2404.16678},
        archivePrefix={arXiv},
        primaryClass={cs.CV}
  }