new_rl_dataset

该数据集使用LeRobot创建，包含机器人相关的观察状态、动作、奖励等信息。具体包括18维的观察状态、4维的动作（delta_x_ee、delta_y_ee、delta_z_ee、gripper_delta）、奖励信号、完成标志等。此外，还包含前视和腕部摄像头拍摄的视频数据，分辨率为128x128，帧率为10fps。数据集共包含60个episodes，3724帧，120个视频。

原始信息汇总

数据集概述

基本信息

• 许可证: Apache-2.0
• 任务类别: 机器人学
• 标签: LeRobot
• 代码库版本: v2.1

数据集结构

• 总集数: 60
• 总帧数: 3724
• 总任务数: 1
• 总视频数: 120
• 总块数: 1
• 块大小: 1000
• 帧率: 10 fps
• 数据路径: 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
  • 形状: [18]
• action:
  • 数据类型: float32
  • 形状: [4]
  • 名称: delta_x_ee, delta_y_ee, delta_z_ee, gripper_delta
• next.reward:
  • 数据类型: float32
  • 形状: [1]
• next.done:
  • 数据类型: bool
  • 形状: [1]
• complementary_info.discrete_penalty:
  • 数据类型: float32
  • 形状: [1]
  • 名称: discrete_penalty
• observation.images.front:
  • 数据类型: video
  • 形状: [3, 128, 128]
  • 名称: channels, height, width
  • 视频信息:
    • 高度: 128
    • 宽度: 128
    • 编解码器: av1
    • 像素格式: yuv420p
    • 是否为深度图: false
    • 帧率: 10 fps
    • 通道数: 3
    • 是否有音频: false
• observation.images.wrist:
  • 数据类型: video
  • 形状: [3, 128, 128]
  • 名称: channels, height, width
  • 视频信息:
    • 高度: 128
    • 宽度: 128
    • 编解码器: av1
    • 像素格式: yuv420p
    • 是否为深度图: false
    • 帧率: 10 fps
    • 通道数: 3
    • 是否有音频: false
• timestamp:
  • 数据类型: float32
  • 形状: [1]
• frame_index:
  • 数据类型: int64
  • 形状: [1]
• episode_index:
  • 数据类型: int64
  • 形状: [1]
• index:
  • 数据类型: int64
  • 形状: [1]
• task_index:
  • 数据类型: int64
  • 形状: [1]

分割信息

• 训练集: 0:60

引用信息

• BibTeX: [More Information Needed]