FreeStyle: Free Lunch for Text-guided Style Transfer using Diffusion Models

Feihong He, Gang Li, Fuhui Sun, Mengyuan Zhang, Lingyu Si +2 more
1/28/2024
cs.CVeess.IV

Official code: FreeStyleFreeLunch/FreeStyle (132 stars, last commit 2024-05-21).

Abstract

The rapid development of generative diffusion models has significantly advanced the field of style transfer. However, most current style transfer methods based on diffusion models typically involve a slow iterative optimization process, e.g., model fine-tuning and textual inversion of style concept. In this paper, we introduce FreeStyle, an innovative style transfer method built upon a pre-trained large diffusion model, requiring no further optimization. Besides, our method enables style transfer only through a text description of the desired style, eliminating the necessity of style images. Specifically, we propose a dual-stream encoder and single-stream decoder architecture, replacing the conventional U-Net in diffusion models. In the dual-stream encoder, two distinct branches take the content image and style text prompt as inputs, achieving content and style decoupling. In the decoder, we further modulate features from the dual streams based on a given content image and the corresponding style text prompt for precise style transfer. Our experimental results demonstrate high-quality synthesis and fidelity of our method across various content images and style text prompts. Compared with state-of-the-art methods that require training, our FreeStyle approach notably reduces the computational burden by thousands of iterations, while achieving comparable or superior performance across multiple evaluation metrics including CLIP Aesthetic Score, CLIP Score, and Preference. We have released the code at: https://github.com/FreeStyleFreeLunch/FreeStyle.

View on arXivView PDF

Code Implementations(1)

FreeStyle : Free Lunch for Text-guided Style Transfer using Diffusion Models

13211PythonFeb 2, 20242 years ago

Similar Papers

Cite this paper

@article{he2024freestyle,
  title  = {FreeStyle: Free Lunch for Text-guided Style Transfer using Diffusion Models},
  author = {Feihong He and Gang Li and Fuhui Sun and Mengyuan Zhang and Lingyu Si and Xiaoyan Wang and Li Shen},
  year   = {2024},
  eprint = {2401.15636},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.15636v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2401.15636.svg)](https://codeofpaper.com/paper/2401.15636)

Discussion