A3-MSI-datasets

# A3-MSI Datasets 本数据集用于论文 ***A3-MSI***（Asymmetric, Autoencoded, Auxiliary-guided Multimodal Seismic Inversion）的地震速度反演测试与泛化评估。A3-MSI 从两类输入反演地下地震速度模型：**走时图（traveltime map, TT）** 作为结构流输入，**早至波波形（early-arrival waveform, EAW）** 作为细节流输入；高维波形经自注意力瓶颈自编码器（SAB-AE）压缩后，通过结构主导非对称融合（SDAF）、FiLM 调制与角色感知辅助监督融合到结构中。 ## 数据组成 数据集包含两个子集，`test/` 与 `fanhua/`（泛化 / 鲁棒性测试集），每个子集 **1200 个样本**，以打包的 HDF5 分片存储。 | 子集 | 样本数 | 样本类型 | 说明 | | --- | --- | --- | --- | | `test` | 1200 | fault / inclined / low | 常规测试集 | | `fanhua` | 1200 | Folded / Unconformity | 泛化（分布外）测试集 | ## 目录结构 ``` A3-MSI-datasets/ test/ _packed_h5/ index.json shard_0000.h5 ... shard_0005.h5 fanhua/ _packed_h5/ index.json shard_0000.h5 ... shard_0005.h5 ``` 每个子集均由 `index.json` 描述元信息（分片布局、样本数、类型分布、ID 范围），其下 `shard_0000.h5` ~ `shard_0005.h5` 为数据分片，每片 200 个样本、6 片合计 1200 样本，无压缩。 ## 数据字段 每个 `.h5` 分片按样本索引存储如下字段： | 字段 | 说明 | | --- | --- | | `x` | 输入特征图（如走时图，具体语义随 `input_mode` 而定） | | `w` | 早至波波形（waveform） | | `y` | 真实速度模型（ground-truth label） | | `y_smooth` | 平滑速度模型 | | `sample_type` | 样本类型（如 fault / inclined / low / Folded / Unconformity） | | `sample_name` | 样本名称 | | `g_max` / `t_max` | 可选归一化标量 | ## 使用方法 可通过 GIT Clone 或 ModelScope SDK 下载本数据集。 **GIT Clone** ```bash git clone https://www.modelscope.cn/datasets/Violetljf/A3-MSI-datasets.git ``` **配合官方代码库加载** 数据结构与官方代码库 [violetwe/A3-MSI](https://github.com/violetwe/A3-MSI) 的 `datasets/datasets_sei_3.py` 对齐。将数据根目录指向本数据集（含 `test/` 与 `fanhua/`），并按代码库 README 设置 `--data_layout`（支持 `auto` / `legacy` / `sample_mat_bundle` / `packed_h5`）即可加载，例如： ```bash python test3.py \ --root_path /path/to/A3-MSI-datasets \ --data_layout packed_h5 \ --exp_name lt1 ``` ## 许可证与引用 - License: Apache License 2.0 - 若使用本数据集，请引用 A3-MSI 论文。