pusht_v2_keypoints

该数据集是使用LeRobot创建的，涉及2D指针机器人。数据集包含206个episodes，25650帧，1个任务，无视频。数据以parquet文件格式存储，包含多种特征，如观察状态（x, y坐标）、动作（x, y坐标）、奖励、成功标志以及各种索引（时间戳、帧索引、episode索引等）。数据集的帧率为10fps，所有数据均用于训练。

原始信息汇总

数据集概述

基本信息

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

数据集详情

• 创建工具: LeRobot
• 代码库版本: v2.0
• 机器人类型: 2D指针 (2d pointer)
• 总集数: 206
• 总帧数: 25650
• 总任务数: 1
• 总视频数: 0
• 总块数: 1
• 块大小: 1000
• 帧率 (fps): 10

数据分割

• 训练集: 0:206

数据路径

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

特征描述

• observation.state:
  • 数据类型: float32
  • 形状: [2]
  • 名称: ["x", "y"]
• action:
  • 数据类型: float32
  • 形状: [2]
  • 名称: ["x", "y"]
• next.reward:
  • 数据类型: float32
  • 形状: [1]
  • 名称: null
• next.success:
  • 数据类型: bool
  • 形状: [1]
  • 名称: null
• observation.environment_state:
  • 数据类型: float32
  • 形状: [16]
  • 名称: "keypoints"
• timestamp:
  • 数据类型: float32
  • 形状: [1]
  • 名称: null
• frame_index:
  • 数据类型: int64
  • 形状: [1]
  • 名称: null
• episode_index:
  • 数据类型: int64
  • 形状: [1]
  • 名称: null
• index:
  • 数据类型: int64
  • 形状: [1]
  • 名称: null
• task_index:
  • 数据类型: int64
  • 形状: [1]
  • 名称: null