UR5_RTDE_SpaceMouse_EE_pick_and_place_object_Easy_ALL

这是一个机器人操作数据集，由LeRobot工具创建，专门用于UR5 RTDE机器人执行拾放任务（具体任务为UR5_RTDE_SpaceMouse_EE_pick_and_place_object_Easy_ALL）。数据集包含82个训练片段，总计20697帧，帧率为15fps。数据特征包括：动作数据（以7维浮点数组表示末端执行器位姿的增量变化和夹爪命令）、状态观测数据（以14维浮点数组表示，包括关节位置、末端执行器位姿、夹爪状态和命令）、以及两个摄像头的图像观测数据（提供分辨率为480x640的RGB视频，编码为h264，格式为yuv420p）。此外，数据集还包含时间戳、帧索引、片段索引、任务索引等元数据。数据以Parquet文件存储，视频以MP4文件存储，总数据大小约100MB，视频大小约200MB。该数据集适用于机器人学习研究，特别适合用于模仿学习或强化学习任务。This is a robot manipulation dataset created by the LeRobot tool for the UR5 RTDE robot performing pick-and-place tasks (specifically UR5_RTDE_SpaceMouse_EE_pick_and_place_object_Easy_ALL). The dataset includes 82 training episodes, totaling 20,697 frames at a frame rate of 15fps. Data features include: actions (a 7-dimensional float array representing incremental changes in end-effector pose and gripper commands), state observations (a 14-dimensional float array including joint positions, end-effector pose, gripper status, and commands), image observations from two cameras (RGB videos with a resolution of 480x640, encoded as h264 in yuv420p format), and metadata such as timestamps, frame indices, episode indices, and task indices. The data is stored in Parquet files, with videos in MP4 format; the total data size is approximately 100MB, and the video size is about 200MB. The dataset is suitable for robotics learning research, particularly for imitation learning or reinforcement learning tasks.

原始信息汇总

数据集概述

该数据集是一个面向机器人操作任务的数据集，名为 UR5_RTDE_SpaceMouse_EE_pick_and_place_object_Easy_ALL，由 LeRobot 工具创建。

许可证

• 许可证类型：Apache-2.0

数据集结构

数据集共包含 82 个 episode，总计 20697 帧，对应 1 个任务。数据文件总大小约 100 MB，视频文件总大小约 200 MB。帧率为 15 FPS。

• 分割：所有数据（0:82）均用于训练。

特征

数据集包含以下特征字段：

• action：动作数据，数据类型为 float32，形状为 [7]，包含以下维度：

  • delta_ee_pose.x
  • delta_ee_pose.y
  • delta_ee_pose.z
  • delta_ee_pose.rx
  • delta_ee_pose.ry
  • delta_ee_pose.rz
  • gripper_cmd_bin

• observation.state：观测状态，数据类型为 float32，形状为 [14]，包含以下维度：

  • joint_1.pos 至 joint_6.pos（6 个关节位置）
  • ee_pose.x 至 ee_pose.rz（6 个末端执行器位姿参数）
  • gripper_state_norm
  • gripper_cmd_bin

• observation.images：包含两个摄像头图像：

  • camera_1：视频数据，分辨率 480x640，3 通道，H264 编码，帧率 15 FPS
  • camera_2：视频数据，分辨率 480x640，3 通道，H264 编码，帧率 15 FPS

• timestamp：时间戳，float32，形状 [1]

• frame_index：帧索引，int64，形状 [1]

• episode_index：episode 索引，int64，形状 [1]

• index：索引，int64，形状 [1]

• task_index：任务索引，int64，形状 [1]

机器人类型

• 机器人类型：ur5_rtde