HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Plug-and-Play Diffusion Features for Text-Driven Image-to-Image Translation

Narek Tumanyan; Michal Geyer; Shai Bagon; Tali Dekel

Plug-and-Play Diffusion Features for Text-Driven Image-to-Image Translation

Abstract

Large-scale text-to-image generative models have been a revolutionary breakthrough in the evolution of generative AI, allowing us to synthesize diverse images that convey highly complex visual concepts. However, a pivotal challenge in leveraging such models for real-world content creation tasks is providing users with control over the generated content. In this paper, we present a new framework that takes text-to-image synthesis to the realm of image-to-image translation -- given a guidance image and a target text prompt, our method harnesses the power of a pre-trained text-to-image diffusion model to generate a new image that complies with the target text, while preserving the semantic layout of the source image. Specifically, we observe and empirically demonstrate that fine-grained control over the generated structure can be achieved by manipulating spatial features and their self-attention inside the model. This results in a simple and effective approach, where features extracted from the guidance image are directly injected into the generation process of the target image, requiring no training or fine-tuning and applicable for both real or generated guidance images. We demonstrate high-quality results on versatile text-guided image translation tasks, including translating sketches, rough drawings and animations into realistic images, changing of the class and appearance of objects in a given image, and modifications of global qualities such as lighting and color.

Code Repositories

michalgeyer/pnp-diffusers
pytorch
Mentioned in GitHub
thu-cvml/texturediffusion
pytorch
Mentioned in GitHub
Shilin-LU/TF-ICON
pytorch
Mentioned in GitHub
MichalGeyer/plug-and-play
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
text-based-image-editing-on-pie-benchDDIM Inversion+Plug-and-Play
Background LPIPS: 113.46
Background PSNR: 22.28
CLIPSIM: 25.41
Structure Distance: 28.22

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Plug-and-Play Diffusion Features for Text-Driven Image-to-Image Translation | Papers | HyperAI