README.md

---
license: cc-by-nc-4.0
language:
- en
- zh
tags:
- seismology
- ground-motion
- StyleGAN
- earthquake-engineering
- generative-model
---

# UGM Models

[UGM (Utilities for Ground Motions)]的预训练模型文件仓库。

## 文件

| 文件 | 大小 | 用途 |
|------|------|------|
| `KNET-StyleGAN-fid1.730.pkl` | ~309 MB | StyleGAN3 条件生成器（KNET 数据集训练，FID 1.73），用于地震动时频谱生成 |
| `vgg16.pt` | ~528 MB | TorchScript 格式的 VGG16 LPIPS 感知损失模型，用于潜空间反演 |

## 下载

```bash
# 方式一：UGM 内置脚本（推荐）
pip install -e ".[hf]"
python -m ugm.download_models

# 方式二：huggingface_hub
python -c "from huggingface_hub import snapshot_download; snapshot_download('xzk8559/UGM', local_dir='~/.cache/ugm/models')"
```

## 许可

模型权重遵循 **CC BY-NC 4.0**（署名-非商业性使用）。代码仓库见 UGM 主仓库。
`vgg16.pt` 原始来源为 NVIDIA StyleGAN2 的 LPIPS 权重（NVIDIA 许可）。