arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.09477cs.CVcs.GRcs.LG
收藏

UniTune:通过在单张图像上微调扩散模型实现文本驱动的图像编辑

UniTune: Text-Driven Image Editing by Fine Tuning a Diffusion Model on a Single Image

Google Research(谷歌研究院)

Dani Valevski, Matan Kalman, Eyal Molad, Eyal Segalis, Yossi Matias, Yaniv Leviathan

AI总结:

本文提出UniTune方法,通过在单张图像上微调扩散模型并结合加噪初始化和细节插值,实现无需掩码或草图的高保真文本驱动图像编辑,支持同一图像多次编辑及显著视觉变化。

摘要:

文本驱动的图像生成方法近年来展现出令人印象深刻的效果,使普通用户能够通过提供文本描述生成高质量图像。然而,对现有图像进行类似编辑的能力仍然难以实现。文本驱动的图像编辑方法通常需要编辑掩码,难以处理需要显著视觉变化的编辑,并且不易保留被编辑部分的特定细节。在本文中,我们观察到图像生成模型可以仅通过在单张图像上进行微调就转换为图像编辑模型。我们还表明,在采样前用基础图像的加噪版本初始化随机采样器,并在采样后从基础图像中插值相关细节,可以进一步提高编辑操作的质量。结合这些观察,我们提出了UniTune,一种新颖的图像编辑方法。UniTune以任意图像和文本编辑描述作为输入,在保持对输入图像高保真度的同时执行编辑。UniTune不需要额外输入,如掩码或草图,并且可以在同一图像上执行多次编辑而无需重新训练。我们使用Imagen模型在一系列不同用例中测试了我们的方法。我们证明该方法具有广泛的适用性,能够执行令人惊讶的广泛表现性编辑操作,包括以前不可能实现的、需要显著视觉变化的编辑。

英文摘要:

Text-driven image generation methods have shown impressive results recently, allowing casual users to generate high quality images by providing textual descriptions. However, similar capabilities for editing existing images are still out of reach. Text-driven image editing methods usually need edit masks, struggle with edits that require significant visual changes and cannot easily keep specific details of the edited portion. In this paper we make the observation that image-generation models can be converted to image-editing models simply by fine-tuning them on a single image. We also show that initializing the stochastic sampler with a noised version of the base image before the sampling and interpolating relevant details from the base image after sampling further increase the quality of the edit operation. Combining these observations, we propose UniTune, a novel image editing method. UniTune gets as input an arbitrary image and a textual edit description, and carries out the edit while maintaining high fidelity to the input image. UniTune does not require additional inputs, like masks or sketches, and can perform multiple edits on the same image without retraining. We test our method using the Imagen model in a range of different use cases. We demonstrate that it is broadly applicable and can perform a surprisingly wide range of expressive editing operations, including those requiring significant visual changes that were previously impossible.

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

↑