UPDF AI

DiffuseST: Unleashing the Capability of the Diffusion Model for Style Transfer

Ying Hu,Chenyi Zhuang,Pan Gao

2024 · DOI: 10.1145/3696409.3700201
ACM Multimedia Asia · 12 Citations

TLDR

This work proposes a novel and training-free approach for style transfer, combining textual embedding with spatial features and separating the injection of content or style, and adopts the BLIP-2 encoder to extract the textual representation of the style image.

Abstract

Style transfer aims to fuse the artistic representation of a style image with the structural information of a content image. Existing methods train specific networks or utilize pre-trained models to learn content and style features. However, they rely solely on textual or spatial representations that are inadequate to achieve the balance between content and style. In this work, we propose a novel and training-free approach for style transfer, combining textual embedding with spatial features and separating the injection of content or style. Specifically, we adopt the BLIP-2 encoder to extract the textual representation of the style image. We utilize the DDIM inversion technique to extract intermediate embeddings in content and style branches as spatial features. Finally, we harness the step-by-step property of diffusion models by separating the injection of content and style in the target branch, which improves the balance between content preservation and style fusion. Various experiments have demonstrated the effectiveness and robustness of our proposed DiffuseST for achieving balanced and controllable style transfer results, as well as the potential to extend to other tasks. Code is available at https://github.com/I2-Multimedia-Lab/DiffuseST.