metadata
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 感知损失模型，用于潜空间反演 |
下载
# 方式一：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 许可）。