satclip-terrain / README.md
latterworks's picture
Upload README.md with huggingface_hub
d854254 verified
metadata
tags:
  - terrain-generation
  - satclip
  - diffusion
  - heightmap
  - procedural
license: mit

SatCLIP Terrain Generator

Generate realistic terrain heightmaps conditioned on geographic coordinates via SatCLIP embeddings.

Usage

from huggingface_hub import hf_hub_download
import torch

ckpt = hf_hub_download("latterworks/satclip-terrain", "model.pt")
weights = torch.load(ckpt)

# Requires satclip + model architecture from training code
model.vae.load_state_dict(weights['vae'])
model.prior.load_state_dict(weights['prior'])
model.refiner.load_state_dict(weights['refiner'])

Training

  • VAE + diffusion prior on SRTM 90m elevation data
  • Colorado/Utah region tiles
  • Conditioned on microsoft/SatCLIP-ResNet18-L40 location embeddings