g1_wbt_inspire_put_clothes_into_washing_machine_maincamonly
收藏资源简介:
这是一个Apache TsFile格式转换的数值时间序列数据集,源自Unitree G1机器人全身遥操作数据集(使用Inspire手部)。任务为“将篮子中的衣物放入洗衣机”。数据集包含457个训练片段,总计119,489帧,涵盖了机器人状态观测(如末端执行器姿态、手部状态、机器人关节配置)和动作数据(如目标末端执行器姿态、手部命令、期望关节位置),采样率为30 fps。数据以TsFile格式存储,包含125列,其中时间、片段索引和任务索引作为标签列,其余为数值字段列。视频特征(两个480x640 AV1摄像头流)未包含在此转换中,但可在源数据集中获取。
This is a numerical time series dataset in Apache TsFile format, converted from the Unitree G1 robot full-body teleoperation dataset (using Inspire hands). The task is putting clothes from a basket into a washing machine. The dataset includes 457 training segments, totaling 119,489 frames, covering robot state observations (such as end-effector poses, hand states, robot joint configurations) and action data (such as target end-effector poses, hand commands, desired joint positions), with a sampling rate of 30 fps. Data is stored in TsFile format, containing 125 columns, where time, segment index, and task index serve as label columns, and the rest are numerical field columns. Video features (two 480x640 AV1 camera streams) are not included in this conversion but are available in the source dataset.
数据集概述
- 数据集名称: Unitree G1 Put Clothes into Washing Machine (TsFile)
- 地址: https://huggingface.co/datasets/zjt24/g1_wbt_inspire_put_clothes_into_washing_machine_maincamonly
- 许可证: Apache-2.0
- 任务类别: 机器人(Robotics)
- 标签: LeRobot, tsfile, time-series
- 模态: 时间序列(Time-series)
- 机器人类型: Unitree G1 (unitree_g1)
- 任务描述: 将篮子里的衣服放入洗衣机
- 采样率: 30 fps
- 数据划分: 训练集(train),包含第0至456集,共457个片段(episodes)
数据集规模
| 项目 | 数量 | 说明 |
|---|---|---|
| 源片段(Episodes) | 457 | 片段索引 0 至 456 |
| 源帧/ TsFile 行数 | 119,489 | 未删除任何数值帧行 |
| 源 Parquet 帧文件 | 1 | G1_WB_Dex5_Put_Clothes_into_Washing_Machine/data/chunk-000/file-000.parquet |
| 源视频文件 | 17 | cam_0 和 cam_1 AV1 流,本数据集未上传 |
| 转换后的 TsFiles | 1 | data/g1_wbt_inspire_put_clothes_into_washing_machine_maincamonly.tsfile |
| TsFile 列数 | 125 | 包含 Time、2 个 TAG 列和 122 个 FIELD 列 |
TsFile 模式(Schema)
TsFile 表名: g1_wbt_inspire_put_clothes_into_washing_machine_maincamonly
| 角色 | 列名 | 描述 |
|---|---|---|
| TIME | Time |
整数毫秒,round(timestamp * 1000),每个片段内重新计数 |
| TAG | episode_index, task_index |
原始 LeRobot 标识符,共同定义逻辑设备 |
| FIELD | frame_index, sample_index |
帧位置和重命名的源 index |
| FIELD | observation_state_ee_state_0..11 |
左右末端执行器位姿(通过正向运动学计算,包含腰部贡献) |
| FIELD | observation_state_hand_state_0..11 |
当前 Inspire 手部手指状态,每只手6个值 |
| FIELD | observation_state_robot_q_current_0..35 |
当前机器人配置:根部位置和四元数,以及29个关节位置 |
| FIELD | action_ee_action_0..11 |
目标左右末端执行器位姿(包含腰部运动) |
| FIELD | action_hand_cmd_0..11 |
命令的 Inspire 手部手指动作,每只手6个值 |
| FIELD | action_robot_q_desired_0..35 |
期望的根部位姿和29个关节位置 |
- 36 维机器人配置向量:元素 0-2 为根部位置,元素 3-6 为根部四元数 (w, x, y, z),元素 7-35 为机器人关节位置。
- 12 维 Inspire 手部向量:每只手包含食指、中指、无名指、小指、拇指开合、拇指侧倾值,范围 0.0(张开)至 1.0(闭合)。
转换说明
- 单个源 Parquet 帧文件转换为一个 TsFile。
episode_index和task_index保持为 TAG 列,可通过WHERE episode_index = N选择单个片段。 - 固定大小的向量列按行优先顺序展平,保留完整源名称,
.替换为_,并附加基于零的元素索引。浮点向量存储为单精度 FLOAT。 - 源
timestamp列被省略,因为以声明的毫秒精度,可由Time / 1000秒表示。源index重命名为sample_index;frame_index保留。 - 相机特征
observation.images.cam_0和observation.images.cam_1被有意排除;帧索引和时间戳保持与源视频对齐。 - 源元数据镜像在
meta/下;meta/info.json将data_path更新为 TsFile 路径,并添加tsfile_conversion信息。
读取数据示例
python from tsfile import TsFileReader
path = "data/g1_wbt_inspire_put_clothes_into_washing_machine_maincamonly.tsfile" table = "g1_wbt_inspire_put_clothes_into_washing_machine_maincamonly"
reader = TsFileReader(path) with reader.query_table( table, ["episode_index", "task_index", "frame_index", "action_ee_action_0"], batch_size=1024, ) as result: batch = result.read_arrow_batch() if batch is not None: print(batch.to_pandas().head())
原始来源
- 原始数据集: unitreerobotics/G1_WBT_Inspire_Put_Clothes_into_Washing_Machine_MainCamOnly
- 源版本: c0a5fb0992a0f2a2b9df3493d27c2d670a4b1c36
- 源作者: unitreerobotics
- 框架: LeRobot



