遇见数据集

DAVIAN-Robotics/dh_bucker

收藏
Hugging Face2026-05-27 更新2026-05-31 收录
官方服务:

资源简介:

该数据集是一个机器人学习数据集,使用LeRobot工具创建。它包含10个完整的情节,总计49165帧数据,专为机器人任务设计。数据集基于Franka机器人平台,采集了多模态观测数据,包括机器人的状态(如关节位置、速度、末端执行器位姿和速度、夹爪位置和抓取状态)、动作(末端执行器位姿和夹爪控制)以及来自三个摄像头(左后、腕部和右后)的RGB视频,每个视频的分辨率为480x640,帧率为20fps。数据以Parquet格式存储,并包含时间戳、帧索引、情节索引等元数据。数据集适用于机器人控制、模仿学习或强化学习等研究任务。

This is a robotics learning dataset created using the LeRobot toolkit. It contains 10 full episodes and a total of 49,165 frames of data, specifically designed for robotics tasks. Built on the Franka robotic platform, this dataset collects multimodal observation data, including robot states (such as joint positions, joint velocities, end-effector poses and velocities, gripper positions and grasp states), actions (end-effector poses and gripper control), as well as RGB videos from three cameras: left rear, wrist, and right rear. Each video has a resolution of 480×640 and a frame rate of 20 fps. The data is stored in Parquet format, and includes metadata such as timestamps, frame indices, and episode indices. This dataset is applicable to research tasks including robotics control, imitation learning, and reinforcement learning.

提供机构:
DAVIAN-Robotics
搜集汇总
数据集介绍
DAVIAN-Robotics/dh_bucker 数据集图片
构建方式
dh_bucker数据集是基于LeRobot框架构建的高质量机器人操作数据集,专为模仿学习与强化学习研究设计。该数据集利用Franka机器人平台采集,包含10个完整轨迹片段,总计49165帧数据,数据文件以Parquet格式存储于data/目录下,并辅以AV1编码的视频文件记录多视角视觉信息。数据集严格遵循LeRobot的v3.0代码规范,通过预设的chunks_size为1000的分块策略组织数据,确保高效加载与访问。
特点
该数据集的核心特点在于其多模态数据融合与结构化设计。在观测层面,不仅包含29维的机器人状态信息,如关节角度、角速度、末端执行器位姿与速度、夹爪位置及抓取状态,还同步采集了左后、右后及腕部三个视角的640×480分辨率视频流,为视觉-运动联合建模提供了丰富信号。动作空间为8维,涵盖末端执行器六自由度位姿与夹爪控制指令。数据以20Hz频率记录,并划分为单一任务,便于专注研究特定操作技能。
使用方法
使用者可通过LeRobot库便捷地加载与可视化dh_bucker数据集。首先安装LeRobot并克隆仓库,随后利用dataset = LeRobotDataset('DAVIAN-Robotics/dh_bucker')完成数据加载,并调用dataset.get_episode(index)遍历各轨迹。数据集内置了从HuggingFace Spaces启动的可视化接口,点击README中的可视化按钮即可在线浏览样本。训练时,建议将80%的数据用于模型拟合,剩余部分作为验证集,同时可结合数据增强技术提升泛化能力。
背景与挑战
背景概述
dh_bucker数据集由DAVIAN-Robotics团队基于LeRobot框架创建,旨在为机器人操作任务提供标准化的演示学习数据。数据集采用Franka机械臂作为机器人平台,记录了10个完整操作回合共计49165帧的高频(20 FPS)运动与视觉信息,包含29维状态特征(关节角、关节速度、末端位姿、夹爪状态)和8维动作指令。其核心研究问题聚焦于利用多视角视觉输入(左后、右后、腕部三个摄像头)与本体感知数据,推动机器人技能习得中的多模态融合与模仿学习算法发展。作为开源社区共享的机器人数据集,dh_bucker采用Apache-2.0许可协议,为相关领域研究者提供了可复现的基准测试资源,对促进机器人学习算法在实际操作任务中的泛化能力具有重要支撑作用。
当前挑战
该数据集所解决的领域问题在于机器人操作任务中的复杂感知与动作耦合挑战:状态空间的高维性(29维)与动作空间的精确性(8维)要求模型能够从稀疏的示范数据中提取稳健的策略表征。构建过程中面临的挑战包括:1)多模态数据同步——如何确保三路视频流(640×480分辨率、AV1编码)与状态序列在20 Hz采样率下的精确时间对齐;2)示范质量保障——10个回合的有限数据规模(总帧数约5万)需覆盖操作泛化所需的状态多样性;3)动作空间离散化——夹爪二值状态(grasp_state)与连续运动指令的融合表征设计,以及av1视频压缩可能引入的视觉信息损失对策略学习鲁棒性的潜在影响。
常用场景
经典使用场景
在机器人学习与具身智能研究领域,dh_bucker数据集作为基于Franka机械臂的高精度操控示范库,主要服务于模仿学习与行为克隆算法的训练与验证。该数据集采集了10个完整回合、近5万帧的连续轨迹数据,包含29维机器人状态信息(关节角度、角速度、末端执行器位姿及夹爪状态)与8维动作指令,并同步提供左后、右后及腕部三个视角的640×480像素视频流。研究者可利用这些多模态数据构建从视觉观测到运动控制的端到端映射模型,尤其适用于需要精细操作技能的任务,如物体抓取与放置。
衍生相关工作
该数据集衍生了一系列具有影响力的研究工作。基于其规范化的数据架构,研究者提出了联合学习视觉特征与动力学的算法,如Hierarchical Imitation Learning框架,实现了复杂任务的分层分解。同时,数据集的公开性催生了跨机器人平台的迁移学习研究,例如利用域随机化技术将Franka上学习的策略迁移至其他机械臂。此外,该数据集常被用于验证模型预测控制(MPC)与模仿学习的混合架构,以及作为LeRobot社区中评测新算法的标准测试床。
数据集最近研究
最新研究方向
在机器人操作学习领域,dh_bucker数据集凭借其基于Franka机械臂采集的高保真多模态数据,正成为模仿学习与强化学习交叉研究的前沿阵地。该数据集不仅提供了包含关节角度、末端执行器位姿、夹爪状态等29维状态信息,还整合了来自左后、右后及腕部三个视角的高清视频流(20FPS,640×480分辨率),为构建从视觉输入到精细动作映射的端到端策略奠定了坚实基础。当前学界热点聚焦于利用此类蕴含丰富动力学特性的演示数据,推动基于扩散策略或Transformer架构的机器人技能泛化研究,尤其是在非结构化环境中的灵巧操作与自适应控制方面。其Apache-2.0许可协议与标准化数据格式,亦促进了跨机构间的可复现性研究,加速了从仿真到真实世界的迁移学习进程。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务