so101-puzzle

该数据集由LeRobot创建，主要用于机器人领域。数据集包含12个episodes，19334帧，1个任务。数据以parquet格式存储，视频以mp4格式存储。数据集结构包含多个特征，如动作（包括肩部、肘部、腕部和夹持器的位置）、观察状态（与动作相同的关节位置）、图像（手腕和前方视角，分辨率为480x640，30fps）、时间戳、帧索引、episode索引等。数据集遵循apache-2.0许可。

原始信息汇总

数据集概述

基本信息

• 数据集名称: so101-puzzle
• 创建工具: LeRobot (https://github.com/huggingface/lerobot)
• 许可证: Apache-2.0
• 任务类别: 机器人学
• 标签: LeRobot

数据集规模

• 总情节数: 12
• 总帧数: 19334
• 总任务数: 1
• 数据文件大小: 100 MB
• 视频文件大小: 200 MB
• 帧率: 30 FPS
• 数据块大小: 1000

数据划分

• 训练集: 包含全部12个情节

数据结构

数据存储为Parquet文件，路径模式为：data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet 视频存储为MP4文件，路径模式为：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

腕部图像观测

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

前部图像观测

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

元数据

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

可视化

可通过以下链接可视化数据集：https://huggingface.co/spaces/lerobot/visualize_dataset?path=aivanni/so101-puzzle

引用信息

• 主页: 未提供
• 论文: 未提供
• BibTeX引用: 未提供