Geosynth OSM

🚀 拡散モデル (diffusers)

このモデルは、OpenStreetMapの画像から衛星画像を合成するControlNetベースのモデルです。ベースとなるStable Diffusionモデルは、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
import torch
from PIL import Image

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

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

pipe = StableDiffusionControlNetPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", controlnet=controlnet)
pipe = pipe.to("cuda:0")

# 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ライセンスの下で提供されています。