diff --git a/README.md b/README.md index 67cf43d..d6ce487 100644 --- a/README.md +++ b/README.md @@ -94,15 +94,14 @@ Projector 负责将 VLM 特征维度对齐到 Agent 的 Embedding 维度。 ### 1. 数据结构标准 数据集必须遵循 [Robomimic](https://robomimic.github.io/) 的层级结构: ```text -dataset.hdf5 -├── data/ -│ ├── demo_0/ -│ │ ├── obs/ -│ │ │ ├── agentview_rgb # (T, H, W, 3) uint8 -│ │ │ └── qpos # (T, D) float32 -│ │ ├── actions # (T, D) float32 -│ │ └── language # (Attribute) String 指令 -│ └── ... +episode_0.hdf5 +├── action: Dataset, shape=(700, 16), dtype=float32 +└── observations: Group + ├── images: Group + │ ├── angle: Dataset, shape=(700, 480, 640, 3), dtype=uint8 + │ ├── r_vis: Dataset, shape=(700, 480, 640, 3), dtype=uint8 + │ └── top: Dataset, shape=(700, 480, 640, 3), dtype=uint8 + └── qpos: Dataset, shape=(700, 16), dtype=float32 ``` ### 2. 数据转换工具