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

基于带步长感知与层感知提示的Stable Diffusion实现高真实感艺术风格迁移

Towards Highly Realistic Artistic Style Transfer via Stable Diffusion with Step-aware and Layer-aware Prompt

Zhejiang University(浙江大学);Anhui University(安徽大学);Zhejiang Gongshang University(浙江工商大学)

Zhanjie Zhang, Quanwei Zhang, Huaizhong Lin, Wei Xing, Juncheng Mo, Shuaicheng Huang, Jinheng Xie, Guangyuan Li, Junsheng Luan, Lei Zhao, Dalong Zhang, Lixia Chen

AI总结:

针对现有艺术风格迁移方法真实感不足或内容结构保持差的问题,提出基于预训练扩散模型的LSAST方法,通过步长与层感知提示空间、对应逆推方法及ControlNet分支,生成更真实且保留内容结构的风格化图像。

摘要:

艺术风格迁移旨在将习得的艺术风格迁移到任意内容图像上,生成艺术风格化图像。现有的基于生成对抗网络的方法无法生成高真实感的风格化图像,且总会引入明显的伪影和不协调的图案。近年来,大规模预训练扩散模型为生成高真实感艺术风格化图像开辟了新途径。然而,基于扩散模型的方法通常无法很好地保留输入内容图像的内容结构,会引入一些不必要的内容结构和风格图案。\n为解决上述问题,我们提出了一种新颖的基于预训练扩散模型的艺术风格迁移方法,名为LSAST,它能够生成高真实感的艺术风格化图像,同时很好地保留输入内容图像的内容结构,不会带来明显的伪影和不协调的风格图案。具体而言,我们引入了步长感知与层感知提示空间(Step-aware and Layer-aware Prompt Space),这是一组可学习的提示,能够从艺术作品集中学习风格信息,并动态调整输入图像的内容结构和风格图案。\n为训练我们的提示空间,我们提出了一种新颖的逆推方法,名为步长感知与层感知提示逆推(Step-aware and Layer-aware Prompt Inversion),它能让提示空间学习艺术作品集的风格信息。此外,我们将预训练的ControlNet条件分支注入到我们的LSAST中,进一步提升了框架的内容结构保持能力。大量实验表明,我们提出的方法相比现有最优的艺术风格迁移方法,能够生成更具真实感的艺术风格化图像。

英文摘要:

Artistic style transfer aims to transfer the learned artistic style onto an arbitrary content image, generating artistic stylized images. Existing generative adversarial network-based methods fail to generate highly realistic stylized images and always introduce obvious artifacts and disharmonious patterns. Recently, large-scale pre-trained diffusion models opened up a new way for generating highly realistic artistic stylized images. However, diffusion model-based methods generally fail to preserve the content structure of input content images well, introducing some undesired content structure and style patterns. To address the above problems, we propose a novel pre-trained diffusion-based artistic style transfer method, called LSAST, which can generate highly realistic artistic stylized images while preserving the content structure of input content images well, without bringing obvious artifacts and disharmonious style patterns. Specifically, we introduce a Step-aware and Layer-aware Prompt Space, a set of learnable prompts, which can learn the style information from the collection of artworks and dynamically adjusts the input images' content structure and style pattern. To train our prompt space, we propose a novel inversion method, called Step-ware and Layer-aware Prompt Inversion, which allows the prompt space to learn the style information of the artworks collection. In addition, we inject a pre-trained conditional branch of ControlNet into our LSAST, which further improved our framework's ability to maintain content structure. Extensive experiments demonstrate that our proposed method can generate more highly realistic artistic stylized images than the state-of-the-art artistic style transfer methods.

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

↑