100 lines
4.4 KiB
Plaintext
100 lines
4.4 KiB
Plaintext
<!--Copyright 2022 The HuggingFace Team. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
|
|
the License. You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
|
|
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
specific language governing permissions and limitations under the License.
|
|
-->
|
|
|
|
# Cycle Diffusion
|
|
|
|
## Overview
|
|
|
|
Cycle Diffusion is a Text-Guided Image-to-Image Generation model proposed in [Unifying Diffusion Models' Latent Space, with Applications to CycleDiffusion and Guidance](https://arxiv.org/abs/2210.05559) by Chen Henry Wu, Fernando De la Torre.
|
|
|
|
The abstract of the paper is the following:
|
|
|
|
*Diffusion models have achieved unprecedented performance in generative modeling. The commonly-adopted formulation of the latent code of diffusion models is a sequence of gradually denoised samples, as opposed to the simpler (e.g., Gaussian) latent space of GANs, VAEs, and normalizing flows. This paper provides an alternative, Gaussian formulation of the latent space of various diffusion models, as well as an invertible DPM-Encoder that maps images into the latent space. While our formulation is purely based on the definition of diffusion models, we demonstrate several intriguing consequences. (1) Empirically, we observe that a common latent space emerges from two diffusion models trained independently on related domains. In light of this finding, we propose CycleDiffusion, which uses DPM-Encoder for unpaired image-to-image translation. Furthermore, applying CycleDiffusion to text-to-image diffusion models, we show that large-scale text-to-image diffusion models can be used as zero-shot image-to-image editors. (2) One can guide pre-trained diffusion models and GANs by controlling the latent codes in a unified, plug-and-play formulation based on energy-based models. Using the CLIP model and a face recognition model as guidance, we demonstrate that diffusion models have better coverage of low-density sub-populations and individuals than GANs.*
|
|
|
|
*Tips*:
|
|
- The Cycle Diffusion pipeline is fully compatible with any [Stable Diffusion](./stable_diffusion) checkpoints
|
|
- Currently Cycle Diffusion only works with the [`DDIMScheduler`].
|
|
|
|
*Example*:
|
|
|
|
In the following we should how to best use the [`CycleDiffusionPipeline`]
|
|
|
|
```python
|
|
import requests
|
|
import torch
|
|
from PIL import Image
|
|
from io import BytesIO
|
|
|
|
from diffusers import CycleDiffusionPipeline, DDIMScheduler
|
|
|
|
# load the pipeline
|
|
# make sure you're logged in with `huggingface-cli login`
|
|
model_id_or_path = "CompVis/stable-diffusion-v1-4"
|
|
scheduler = DDIMScheduler.from_pretrained(model_id_or_path, subfolder="scheduler")
|
|
pipe = CycleDiffusionPipeline.from_pretrained(model_id_or_path, scheduler=scheduler).to("cuda")
|
|
|
|
# let's download an initial image
|
|
url = "https://raw.githubusercontent.com/ChenWu98/cycle-diffusion/main/data/dalle2/An%20astronaut%20riding%20a%20horse.png"
|
|
response = requests.get(url)
|
|
init_image = Image.open(BytesIO(response.content)).convert("RGB")
|
|
init_image = init_image.resize((512, 512))
|
|
init_image.save("horse.png")
|
|
|
|
# let's specify a prompt
|
|
source_prompt = "An astronaut riding a horse"
|
|
prompt = "An astronaut riding an elephant"
|
|
|
|
# call the pipeline
|
|
image = pipe(
|
|
prompt=prompt,
|
|
source_prompt=source_prompt,
|
|
image=init_image,
|
|
num_inference_steps=100,
|
|
eta=0.1,
|
|
strength=0.8,
|
|
guidance_scale=2,
|
|
source_guidance_scale=1,
|
|
).images[0]
|
|
|
|
image.save("horse_to_elephant.png")
|
|
|
|
# let's try another example
|
|
# See more samples at the original repo: https://github.com/ChenWu98/cycle-diffusion
|
|
url = "https://raw.githubusercontent.com/ChenWu98/cycle-diffusion/main/data/dalle2/A%20black%20colored%20car.png"
|
|
response = requests.get(url)
|
|
init_image = Image.open(BytesIO(response.content)).convert("RGB")
|
|
init_image = init_image.resize((512, 512))
|
|
init_image.save("black.png")
|
|
|
|
source_prompt = "A black colored car"
|
|
prompt = "A blue colored car"
|
|
|
|
# call the pipeline
|
|
torch.manual_seed(0)
|
|
image = pipe(
|
|
prompt=prompt,
|
|
source_prompt=source_prompt,
|
|
image=init_image,
|
|
num_inference_steps=100,
|
|
eta=0.1,
|
|
strength=0.85,
|
|
guidance_scale=3,
|
|
source_guidance_scale=1,
|
|
).images[0]
|
|
|
|
image.save("black_to_blue.png")
|
|
```
|
|
|
|
## CycleDiffusionPipeline
|
|
[[autodoc]] CycleDiffusionPipeline
|
|
- __call__
|