so101_grasp_dataset

该数据集为HuggingFace LeRobot格式机器人数据集。

原始信息汇总

数据集概述

基本信息

• 数据集名称: so101_grasp_dataset
• 发布平台: Hugging Face
• 许可证: Apache-2.0
• 任务类别: 机器人学
• 相关标签: LeRobot

数据集来源

• 该数据集使用 LeRobot 创建。

数据集结构

• 数据文件格式: Parquet
• 数据文件路径模式: data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
• 视频文件路径模式: videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4
• 配置: 默认配置 (default)，数据文件位于 data/*/*.parquet。

数据集规模

• 总情节数: 2
• 总帧数: 263
• 总任务数: 1
• 数据块大小: 1000
• 数据文件总大小: 100 MB
• 视频文件总大小: 200 MB
• 帧率: 30 FPS
• 数据划分: 训练集包含所有情节 ("train": "0:2")。

数据特征

数据集包含以下特征：

动作

• 数据类型: float32
• 形状: [6]
• 内容: 机械臂关节位置，包括：
  • shoulder_pan.pos
  • shoulder_lift.pos
  • elbow_flex.pos
  • wrist_flex.pos
  • wrist_roll.pos
  • gripper.pos

观测状态

• 数据类型: float32
• 形状: [6]
• 内容: 机械臂关节位置，与动作特征相同。

观测图像（前视）

• 数据类型: video
• 形状: [720, 1280, 3] (高度，宽度，通道)
• 视频信息:
  • 高度: 720 像素
  • 宽度: 1280 像素
  • 编解码器: av1
  • 像素格式: yuv420p
  • 是否为深度图: false
  • 帧率: 30 FPS
  • 通道数: 3
  • 是否包含音频: false

元数据

• 时间戳: float32，形状 [1]
• 帧索引: int64，形状 [1]
• 情节索引: int64，形状 [1]
• 索引: int64，形状 [1]
• 任务索引: int64，形状 [1]

其他信息

• 代码库版本: v3.0
• 机器人类型: so_follower
• 主页: 未提供
• 论文: 未提供
• 引用信息: 未提供