five

push_franka_robotiq

收藏
Hugging Face2026-06-28 更新2026-06-28 收录
下载链接:
https://huggingface.co/datasets/dexsuite/push_franka_robotiq
下载链接
链接失效反馈
官方服务:
资源简介:
这是一个机器人操作数据集,使用LeRobot工具创建,专门用于推操作任务。数据集包含50个训练片段,总计5588帧数据,帧率为20fps。数据以parquet格式存储,视频以mp4格式存储。数据集的特征包括:机械臂状态(如末端执行器位置、姿态、关节位置、速度和扭矩)、夹爪状态(如位置、姿态、关节位置、速度和扭矩)、其他状态(如立方体位置、姿态、目标位置)、相机观测(包括前视和腕部相机的RGB和深度图像,分辨率均为224x224),以及奖励信号、终止标志(如成功、失败、截断)和动作信息。数据集适用于机器人强化学习或模仿学习的研究,特别是涉及视觉和状态反馈的推操作任务。

This is a robot manipulation dataset created using the LeRobot tool, specifically designed for pushing tasks. The dataset contains 50 training episodes, totaling 5588 frames of data at a frame rate of 20fps. Data is stored in parquet format, and videos are stored in mp4 format. Features of the dataset include: robot arm states (such as end-effector position, orientation, joint positions, velocities, and torques), gripper states (such as position, orientation, joint positions, velocities, and torques), other states (such as cube position, orientation, target position), camera observations (including RGB and depth images from front-view and wrist cameras, with a resolution of 224x224), as well as reward signals, termination flags (e.g., success, failure, truncation), and action information. The dataset is suitable for research in robot reinforcement learning or imitation learning, particularly for pushing tasks involving visual and state feedback.
提供机构:
dexsuite
创建时间:
2026-06-28
原始信息汇总

数据集概述

本数据集名为 push_franka_robotiq,由 LeRobot 框架创建,适用于机器人学(robotics)任务。

  • 许可证: Apache-2.0
  • 任务类别: 机器人学
  • 标签: LeRobot
  • 配置: 默认配置,数据文件路径为 data/*/*.parquet

数据集规模

  • 总集数 (Episodes): 50
  • 总帧数 (Frames): 5,588
  • 总任务数 (Tasks): 1
  • 帧率 (FPS): 20
  • 数据分块: 每块含 1,000 帧
  • 数据文件大小: 100 MB
  • 视频文件大小: 200 MB
  • 数据分割: 训练集使用所有 50 集(0:50

数据集结构

  • 任务名称: push
  • 数据路径: data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet
  • 视频路径: videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4

特征详细说明

特征名称 数据类型 形状 说明
obs.state.gripper.qpos float32 (6,) 夹爪关节位置
obs.state.gripper.qtorque float32 (6,) 夹爪关节扭矩
obs.state.gripper.qvel float32 (6,) 夹爪关节速度
obs.state.gripper.tcp_pos float32 (3,) 夹爪 TCP 位置
obs.state.gripper.tcp_quat float32 (4,) 夹爪 TCP 四元数姿态
obs.state.manipulator.eef_pos float32 (1, 3) 机械臂末端执行器位置
obs.state.manipulator.eef_quat float32 (1, 4) 机械臂末端执行器四元数
obs.state.manipulator.qpos float32 (7,) 机械臂关节位置
obs.state.manipulator.qtorque float32 (7,) 机械臂关节扭矩
obs.state.manipulator.qvel float32 (7,) 机械臂关节速度
obs.state.other.action float32 (8,) 其他关节的动作
obs.state.other.cube_pos float32 (3,) 立方体位置
obs.state.other.cube_quat float32 (4,) 立方体四元数
obs.state.other.goal_pos float32 (3,) 目标位置
obs.state.other.last_action float32 (8,) 上一个动作
obs.cameras.front.rgb video (224, 224, 3) 前置摄像头 RGB 视频(AV1 编码,20 FPS)
obs.cameras.front.depth float32 (224, 224) 前置摄像头深度图
obs.cameras.wrist.rgb video (224, 224, 3) 腕部摄像头 RGB 视频(AV1 编码,20 FPS)
obs.cameras.wrist.depth float32 (224, 224) 腕部摄像头深度图
action float32 (8,) 动作指令
reward float32 (1,) 奖励值
truncated bool (1,) 是否截断
success bool (1,) 是否成功
failure bool (1,) 是否失败
timestamp float32 (1,) 时间戳
frame_index int64 (1,) 帧索引
episode_index int64 (1,) 集索引
index int64 (1,) 全局索引
task_index int64 (1,) 任务索引

引用

目前暂无具体的引用信息(BibTeX 待补充)。

二维码
社区交流群
二维码
科研交流群
商业服务