11

该数据集使用LeRobot创建，是一个机器人学数据集，包含一个总时长为403帧的episode（对应30fps）。数据来自so100_follower机器人，记录了一个任务。数据集包含6自由度机械臂的动作和状态观测（肩部平移、肩部升降、肘部弯曲、腕部弯曲、腕部旋转、夹爪位置），以及来自前方和侧面视角的480x640 RGB视频观测。

原始信息汇总

数据集概述

基本信息

• 许可证: Apache-2.0
• 任务类别: 机器人学
• 标签: LeRobot

数据集结构

• 配置名称: default
• 数据文件: data//.parquet

数据集详情

• 总任务数: 1
• 总视频数: 2
• 总帧数: 403
• 总片段数: 1
• 片段大小: 1000
• 帧率: 30 FPS
• 分割: 训练集 (0:1)
• 代码库版本: v2.1
• 机器人类型: so100_follower

数据特征

动作特征

• 数据类型: float32
• 形状: [6]
• 名称: shoulder_pan.pos, shoulder_lift.pos, elbow_flex.pos, wrist_flex.pos, wrist_roll.pos, gripper.pos

观测状态特征

• 数据类型: float32
• 形状: [6]
• 名称: shoulder_pan.pos, shoulder_lift.pos, elbow_flex.pos, wrist_flex.pos, wrist_roll.pos, gripper.pos

前视图像观测特征

• 数据类型: video
• 形状: [480, 640, 3]
• 名称: height, width, channels
• 视频信息:
  • 高度: 480
  • 宽度: 640
  • 编解码器: av1
  • 像素格式: yuv420p
  • 是否为深度图: false
  • 帧率: 30
  • 通道数: 3
  • 包含音频: false

侧视图像观测特征

• 数据类型: video
• 形状: [480, 640, 3]
• 名称: height, width, channels
• 视频信息:
  • 高度: 480
  • 宽度: 640
  • 编解码器: av1
  • 像素格式: yuv420p
  • 是否为深度图: false
  • 帧率: 30
  • 通道数: 3
  • 包含音频: false

其他特征

• 时间戳: float32, 形状 [1]
• 帧索引: int64, 形状 [1]
• 片段索引: int64, 形状 [1]
• 索引: int64, 形状 [1]
• 任务索引: int64, 形状 [1]

数据存储路径

• 数据路径: data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet
• 视频路径: videos/chunk-{episode_chunk:03d}/{video_key}/episode_{episode_index:06d}.mp4

创建信息

• 创建工具: LeRobot (https://github.com/huggingface/lerobot)