so-101-pick-and-place-9

该数据集由LeRobot创建，主要用于机器人技术研究。数据集包含102个episodes，共33145帧数据，帧率为30fps。数据内容包括机器人的动作（6个关节位置）、观测状态（6个关节位置）、来自两个摄像头的图像（480x640分辨率）、时间戳、帧索引、episode索引等。数据以parquet格式存储，总数据量约100MB，视频数据约200MB。

原始信息汇总

数据集概述

基本信息

• 数据集名称: so-101-pick-and-place-9
• 创建工具: LeRobot (https://github.com/huggingface/lerobot)
• 许可协议: Apache-2.0
• 任务类别: 机器人学
• 标签: LeRobot

数据集结构

• 总任务数: 1
• 总情节数: 102
• 总帧数: 33145
• 数据块大小: 1000
• 数据文件总大小: 100 MB
• 视频文件总大小: 200 MB
• 帧率: 30 FPS
• 数据分割: 训练集 (0:102)

数据文件与路径

• 数据文件格式: Parquet
• 数据文件路径模式: data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
• 视频文件路径模式: videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4

特征字段

动作

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

观测状态

• 字段名: observation.state
• 数据类型: float32
• 形状: [6]
• 维度名称: shoulder_pan.pos, shoulder_lift.pos, elbow_flex.pos, wrist_flex.pos, wrist_roll.pos, gripper.pos

观测图像 (camera2)

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

观测图像 (camera3)

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

元数据字段

• 时间戳: timestamp (float32, 形状 [1])
• 帧索引: frame_index (int64, 形状 [1])
• 情节索引: episode_index (int64, 形状 [1])
• 索引: index (int64, 形状 [1])
• 任务索引: task_index (int64, 形状 [1])

其他信息

• 代码库版本: v3.0
• 机器人类型: so_follower
• 主页: [More Information Needed]
• 论文: [More Information Needed]
• 引用格式: [More Information Needed]