arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.01732cs.CVcs.AIcs.GR

使用预训练文本到图像扩散模型进行视频着色

Video Colorization with Pre-trained Text-to-Image Diffusion Models

The Chinese University of Hong Kong(香港中文大学);Caritas Institute of Higher Education(明爱专上学院)

Hanyuan Liu, Minshan Xie, Jinbo Xing, Chengze Li, Tien-Tsin Wong

AI总结:

提出 ColorDiffuser,将预训练文本到图像潜在扩散模型适配于视频着色,并通过颜色传播注意力与交替采样增强帧间一致性,实现优于现有方法的着色效果。

摘要:

视频着色是一项具有挑战性的任务,涉及为灰度帧推断合理且时间一致的颜色。本文提出 ColorDiffuser,这是一种对预训练文本到图像潜在扩散模型的改编,用于视频着色。通过所提出的基于适配器的方法,我们重新调整预训练文本到图像模型,使其接受输入灰度视频帧以及可选的文本描述来进行视频着色。为了增强时间连贯性并保持各帧之间着色的鲜艳度,我们提出了两项新技术:颜色传播注意力和交替采样策略。颜色传播注意力使模型能够基于参考潜在帧优化其着色决策,而交替采样策略通过在生成式扩散采样步骤中交替使用下一个和上一个相邻潜在帧作为参考,来捕捉时空依赖关系。这促进了相邻视频帧之间的双向颜色信息传播,从而改善了各帧之间的颜色一致性。我们在基准数据集上进行了大量实验,结果证明了所提出框架的有效性。评估表明,ColorDiffuser 在视频着色方面取得了最先进的性能,在颜色保真度、时间一致性和视觉质量方面超越了现有方法。

英文摘要:

Video colorization is a challenging task that involves inferring plausible and temporally consistent colors for grayscale frames. In this paper, we present ColorDiffuser, an adaptation of a pre-trained text-to-image latent diffusion model for video colorization. With the proposed adapter-based approach, we repropose the pre-trained text-to-image model to accept input grayscale video frames, with the optional text description, for video colorization. To enhance the temporal coherence and maintain the vividness of colorization across frames, we propose two novel techniques: the Color Propagation Attention and Alternated Sampling Strategy. Color Propagation Attention enables the model to refine its colorization decision based on a reference latent frame, while Alternated Sampling Strategy captures spatiotemporal dependencies by using the next and previous adjacent latent frames alternatively as reference during the generative diffusion sampling steps. This encourages bidirectional color information propagation between adjacent video frames, leading to improved color consistency across frames. We conduct extensive experiments on benchmark datasets, and the results demonstrate the effectiveness of our proposed framework. The evaluations show that ColorDiffuser achieves state-of-the-art performance in video colorization, surpassing existing methods in terms of color fidelity, temporal consistency, and visual quality.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑