arXiv · 2109.14518
Generative Probabilistic Image Colorization
Abstract
We propose Generative Probabilistic Image Colorization, a diffusion-based generative process that trains a sequence of probabilistic models to reverse each step of noise corruption. Given a line-drawing image as input, our method suggests multiple candidate colorized images. Therefore, our method accounts for the ill-posed nature of the colorization problem. We conducted comprehensive experiments investigating the colorization of line-drawing images, report the influence of a score-based MCMC approach that corrects the marginal distribution of estimated samples, and further compare different combinations of models and the similarity of their generated images. Despite using only a relatively small training dataset, we experimentally develop a method to generate multiple diverse colorization candidates which avoids mode collapse and does not require any additional constraints, losses, or re-training with alternative training conditions. Our proposed approach performed well not only on color-conditional image generation tasks using biased initial values, but also on some practical image completion and inpainting tasks.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Chie Furusawa, Shinya Kitaoka, Michael Li, Yuri Odagiri. 2021-09-29. Generative Probabilistic Image Colorization. https://arxiv.org/abs/2109.14518
Cite the original work for its findings. Save a collection to share your selection of sources.