StyleBrush: Style Extraction and Transfer from a Single Image

Wancheng Feng, Wanquan Feng, Dawei Huang, Jiaming Pei, Guangliang Cheng +1 more
8/18/2024
cs.CV

No confident code implementation found for this paper.

Abstract

Stylization for visual content aims to add specific style patterns at the pixel level while preserving the original structural features. Compared with using predefined styles, stylization guided by reference style images is more challenging, where the main difficulty is to effectively separate style from structural elements. In this paper, we propose StyleBrush, a method that accurately captures styles from a reference image and ``brushes'' the extracted style onto other input visual content. Specifically, our architecture consists of two branches: ReferenceNet, which extracts style from the reference image, and Structure Guider, which extracts structural features from the input image, thus enabling image-guided stylization. We utilize LLM and T2I models to create a dataset comprising 100K high-quality style images, encompassing a diverse range of styles and contents with high aesthetic score. To construct training pairs, we crop different regions of the same training image. Experiments show that our approach achieves state-of-the-art results through both qualitative and quantitative analyses. We will release our code and dataset upon acceptance of the paper.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{feng2024stylebrush,
  title  = {StyleBrush: Style Extraction and Transfer from a Single Image},
  author = {Wancheng Feng and Wanquan Feng and Dawei Huang and Jiaming Pei and Guangliang Cheng and Lukun Wang},
  year   = {2024},
  eprint = {2408.09496},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2408.09496v1}
}

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/2408.09496.svg)](https://codeofpaper.com/paper/2408.09496)

Discussion