five

droid_merged_skills_stacking

收藏
Hugging Face2026-05-27 更新2026-05-27 收录
下载链接:
https://huggingface.co/datasets/jellyho/droid_merged_skills_stacking
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集专为机器人学习设计,特别针对堆叠任务。它整合了一个经过语言筛选的DROID技能子集以及新收集的相同技能(堆叠)环境数据。数据集使用标准化的8维向量来表示状态和动作,这些向量包含了关节位置、夹爪位置等信息。整个数据集规模为319个训练片段(episodes)、61511帧(frames)以及301个不同的任务。此外,数据集还提供了视频索引键和任务字符串的示例。在文件组织上,它包含了元数据、实际的数据文件以及相关的视频文件。

This dataset is designed for robot learning, specifically for stacking tasks. It integrates a language-filtered subset of DROID skills with newly collected environment data for the same skill (stacking). The dataset employs standardized 8-dimensional vectors to represent states and actions, which include information such as joint positions and gripper positions. The overall scale of the dataset comprises 319 episodes, 61511 frames, and 301 unique tasks. Additionally, the dataset provides examples of video keys and task strings. In terms of file structure, it includes metadata, data files, and associated video files.
提供机构:
jellyho
创建时间:
2026-05-26
原始信息汇总

数据集概述:Merged DROID Skill Dataset: stacking

简介
该数据集将经过语言筛选的DROID技能子集与新收集的相同技能的环境数据合并而成,专注于“stacking(堆叠)”技能。基于LeRobot v2.1代码库构建。

许可
Apache-2.0

任务类别
机器人学(robotics)

标签
LeRobot, DROID, robotics, skill-filtered, merged, joint-velocity-action, gripper-position, stacking


数据来源
数据集由以下两个子数据集合并而成:

源数据集 片段数
jellyho/droid_subsets_stacking 299
jellyho/droid_stack_bar 20

状态与动作空间
紧凑的LeRobot训练字段标准化为8维向量:

  • observation.state = [关节位置0..6, 夹爪位置]
  • action = [关节速度0..6, 夹爪位置]

对于原始DROID子集的行,action来源于action.joint_velocity加上action.gripper_position。对于新收集的行,action来源于observation.joint_velocities[:7]加上原始action向量中的夹爪位置。详细的列(如action.joint_positionaction.joint_velocityaction.gripper_positionobservation.state.joint_positionobservation.state.gripper_position)在parquet文件中仍可用。


数据集统计

指标 数值
片段数 (Episodes) 319
帧数 (Frames) 61511
唯一任务数 (Unique tasks) 301
observation.state 形状 [8]
action 形状 [8]
视频键 (Video keys) [observation.images.wrist_left, observation.images.side_view_1_left]
视频数 (Videos) 638

顶部片段任务示例

任务 片段数
[Stack the yellow bar on top of the red cube] 20
[Stack the cups from the top cup to the bottom cup] 1
[Pick up the lid and place it on top of the blue pot.] 1
[Move the white mug from on top of the paper towel to the counter] 1
[Remove the cap from the door coat hanging hook and put it on top of the cabinet] 1
[Stack the items on the tablecloth together.] 1
[Fold the white cloth in half, pick up the white cloth and put it on top of the far right throw pillow] 1
[Place the grey object at the top of the stack] 1
[Place the yellow chip on the right on top of the two yellow chips.] 1
[Stack the cups on the table] 1

文件结构

meta/info.json meta/tasks.jsonl meta/episodes.jsonl meta/episodes_stats.jsonl data/chunk-000/episode_.parquet videos/chunk-000/observation.images.wrist_left/episode_.mp4 episodes.csv merge_summary.json

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