Geosynth Canny

🚀 Diffusers模型

這是一個基於ControlNet的模型，可根據Canny邊緣圖像合成衛星圖像。所使用的基礎穩定擴散模型為 stable-diffusion-2-1-base (v2-1_512-ema-pruned.ckpt)，能夠高效地實現圖像到圖像的轉換。

🚀 快速開始

此模型可通過以下兩種方式使用：

• 使用 🧨 diffusers 庫調用
• 使用 controlnet 倉庫中的代碼

模型來源

• 倉庫地址：stable-diffusion
• 相關論文：Adding Conditional Control to Text-to-Image Diffusion Models

💻 使用示例

基礎用法

from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
import torch
from PIL import Image

img = Image.open("canny_example.jpeg")

controlnet = ControlNetModel.from_pretrained("MVRL/GeoSynth-Canny")

scheduler = UniPCMultistepScheduler.from_pretrained("stabilityai/stable-diffusion-2-1-base", subfolder="scheduler")

pipe = StableDiffusionControlNetPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", controlnet=controlnet, scheduler=scheduler)
pipe.enable_xformers_memory_efficient_attention()
pipe.enable_model_cpu_offload()

# generate image
generator = torch.manual_seed(10345340)
image = pipe(
    "Satellite image features a city neighborhood",
    generator=generator,
    image=img,
).images[0]

image.save("generated_city.jpg")

📄 許可證

本模型採用Apache-2.0許可證。

引用信息

BibTeX： [More Information Needed]

APA： [More Information Needed]

更多信息

[More Information Needed]

模型卡片作者

[More Information Needed]

模型卡片聯繫方式

[More Information Needed]