ColorFlow: Retrieval-Augmented Image Sequence Colorization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhuang, Junhao, Ju, Xuan, Zhang, Zhaoyang, Liu, Yong, Zhang, Shiyi, Yuan, Chun, Shan, Ying
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910926971600896
author Zhuang, Junhao
Ju, Xuan
Zhang, Zhaoyang
Liu, Yong
Zhang, Shiyi
Yuan, Chun
Shan, Ying
author_facet Zhuang, Junhao
Ju, Xuan
Zhang, Zhaoyang
Liu, Yong
Zhang, Shiyi
Yuan, Chun
Shan, Ying
contents Automatic black-and-white image sequence colorization while preserving character and object identity (ID) is a complex task with significant market demand, such as in cartoon or comic series colorization. Despite advancements in visual colorization using large-scale generative models like diffusion models, challenges with controllability and identity consistency persist, making current solutions unsuitable for industrial application.To address this, we propose ColorFlow, a three-stage diffusion-based framework tailored for image sequence colorization in industrial applications. Unlike existing methods that require per-ID finetuning or explicit ID embedding extraction, we propose a novel robust and generalizable Retrieval Augmented Colorization pipeline for colorizing images with relevant color references. Our pipeline also features a dual-branch design: one branch for color identity extraction and the other for colorization, leveraging the strengths of diffusion models. We utilize the self-attention mechanism in diffusion models for strong in-context learning and color identity matching. To evaluate our model, we introduce ColorFlow-Bench, a comprehensive benchmark for reference-based colorization. Results show that ColorFlow outperforms existing models across multiple metrics, setting a new standard in sequential image colorization and potentially benefiting the art industry. We release our codes and models on our project page: https://zhuang2002.github.io/ColorFlow/.
format Preprint
id arxiv_https___arxiv_org_abs_2412_11815
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ColorFlow: Retrieval-Augmented Image Sequence Colorization
Zhuang, Junhao
Ju, Xuan
Zhang, Zhaoyang
Liu, Yong
Zhang, Shiyi
Yuan, Chun
Shan, Ying
Computer Vision and Pattern Recognition
Automatic black-and-white image sequence colorization while preserving character and object identity (ID) is a complex task with significant market demand, such as in cartoon or comic series colorization. Despite advancements in visual colorization using large-scale generative models like diffusion models, challenges with controllability and identity consistency persist, making current solutions unsuitable for industrial application.To address this, we propose ColorFlow, a three-stage diffusion-based framework tailored for image sequence colorization in industrial applications. Unlike existing methods that require per-ID finetuning or explicit ID embedding extraction, we propose a novel robust and generalizable Retrieval Augmented Colorization pipeline for colorizing images with relevant color references. Our pipeline also features a dual-branch design: one branch for color identity extraction and the other for colorization, leveraging the strengths of diffusion models. We utilize the self-attention mechanism in diffusion models for strong in-context learning and color identity matching. To evaluate our model, we introduce ColorFlow-Bench, a comprehensive benchmark for reference-based colorization. Results show that ColorFlow outperforms existing models across multiple metrics, setting a new standard in sequential image colorization and potentially benefiting the art industry. We release our codes and models on our project page: https://zhuang2002.github.io/ColorFlow/.
title ColorFlow: Retrieval-Augmented Image Sequence Colorization
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.11815