HuskyMango/5-28
收藏Hugging Face2026-05-29 更新2026-05-31 收录
下载链接:
https://hf-mirror.com/datasets/HuskyMango/5-28
下载链接
链接失效反馈官方服务:
资源简介:
该数据集是一个机器人技术数据集,专为单臂Franka机器人设计,用于机器人学习和控制任务。数据集包含1个完整任务(episode),总计243帧数据,以20帧每秒(fps)的速率采集。数据以parquet格式存储,总数据文件大小为100MB,视频文件大小为200MB。数据集结构包括动作特征(action),如机器人末端执行器的位置(x, y, z)、四元数姿态(qx, qy, qz, qw)和夹爪状态(gripper),以及观察状态特征(observation.state),涵盖机器人7个关节的位置、夹爪状态和深度信息(从depth_0000到depth_0187的x、y、z坐标),用于提供环境感知数据。数据集适用于训练和评估机器人控制算法,支持强化学习或模仿学习应用。
This dataset is a robotics dataset designed for a single-arm Franka robot, intended for robot learning and control tasks. It contains 1 complete task (episode) with a total of 243 frames, captured at a rate of 20 frames per second (fps). The data is stored in parquet format, with a total data file size of 100MB and video file size of 200MB. The dataset structure includes action features, such as robot end-effector position (x, y, z), quaternion orientation (qx, qy, qz, qw), and gripper state, as well as observation state features, covering the positions of 7 robot joints, gripper state, and depth information (x, y, z coordinates from depth_0000 to depth_0187) for environmental perception. The dataset is suitable for training and evaluating robot control algorithms, supporting applications in reinforcement learning or imitation learning.
提供机构:
HuskyMango搜集汇总
数据集介绍

构建方式
该数据集利用LeRobot框架构建,聚焦于机器人操控领域,具体采用单臂Franka机器人平台进行数据采集。数据集仅包含单一任务场景,记录了1个完整回合(episode)的交互历程,共计243帧(frames)时序数据,采样频率为20帧/秒。数据分为两部分:以Parquet格式存储的数值型数据,涵盖机器人关节状态、末端执行器位姿及夹爪开合度等动作指令与状态观测;同时关联MP4格式的视频记录。结构上按chunk与file二级目录组织,并通过info.json文件统一管理元信息,实现了数据与视频的协同映射。
使用方法
在LeRobot框架下,用户可通过Hugging Face提供的可视化工具直接预览数据集。使用时需加载Parquet文件获取数值特征,并配套读取对应索引的视频片段进行多模态对齐。数据按['train']切分全部可用,适合用于模仿学习或强化学习中的策略训练。开发者可通过LeRobot的API按回合或帧索引高效提取数据,结合机器人类型标识与特征命名规则,灵活适配动作预测、状态估计等下游任务,特别适用于单任务小样本场景下的算法快速迭代。
背景与挑战
背景概述
在机器人学习领域,模仿学习与行为克隆等方法依赖于高质量、低延迟的传感器-动作数据对。该数据集“5-28”由HuskyMango团队于近期创建,基于LeRobot框架和Franka单臂机器人平台采集,记录了单次操作任务中243帧、时长约12秒的精细动作序列。数据以20 FPS的采样率同步记录了机器人7个关节角度、夹爪状态以及160个深度点云的三维坐标,构成了从高维感知到关节空间动作的完整映射。尽管规模有限,但该数据集为评估从原始传感器数据到精确动作的端到端学习算法提供了干净的基准,尤其在单任务、低样本场景下的泛化能力研究中具有参考价值。
当前挑战
该数据集的核心挑战在于应对小样本学习中的过拟合问题,仅包含1个episode的243条样本难以支撑复杂深度学习模型的训练。构建过程中,高精度动作与多通道深度信息的时序对齐、20Hz下的实时数据流同步以及约300MB的总存储管理,均对采集系统的可靠性与一致性提出了严苛要求。此外,单一机器人构型与固定任务设定限制了数据集的迁移能力,如何从稀疏标注中提取可泛化的操作策略,仍是该领域亟待解决的瓶颈。
常用场景
经典使用场景
在机器人学习领域,5-28数据集源自LeRobot框架,专为单臂Franka机器人设计,收录了单一任务下的243帧连续动作序列。该数据集最经典的用途在于训练行为克隆(Behavioral Cloning)模型,通过端到端的方式从人类示教数据中学习策略。研究者利用其包含的8维动作指令(末端位姿与夹爪状态)以及丰富的深度点云状态观测(涵盖超过180个三维空间点),构建从感知到动作的直接映射。这一场景为模仿学习提供了高保真的训练素材,使机器人能复现拾取、放置等精密操作,成为验证离线强化学习与示范学习算法的理想基准。
解决学术问题
该数据集有效攻克了机器人操作领域策略泛化与数据高效的学术难题。传统方法常因数据量匮乏而难以学习鲁棒的操作技能,而5-28以标准化的Franka平台与精细的分块存储结构,为小样本模仿学习研究提供了可靠支撑。它解决了从高维状态空间(如深度点云)中提取关键特征以生成连续动作的挑战,推动了端到端策略的收敛性研究。其意义在于建立了一个可复现的微型基准,助力学者分析噪声、延迟与策略偏差等问题,进而提升离线策略学习的稳定性与迁移性,为复杂任务中的技能习得奠定了实验基础。
实际应用
在实际产业场景中,5-28数据集可助力自动化的精密装配与分拣任务。作为Franka机械臂的示教数据,它能够被迁移至实际生产线中,训练机器人执行物料的精准抓取与定向放置。通过深度相机捕获的点云状态,机器人可感知目标物体的三维位姿,进而调整夹爪动作以适应复杂环境。此外,该数据格式与LeRobot框架的兼容性,使其能无缝集成到工业仿真验证平台中,加速从实验室到工厂的技能部署,降低手动编程成本,提升柔性制造的自适应能力。
数据集最近研究
最新研究方向
在机器人学习领域,数据集扮演着驱动算法演进的关键角色。名为“5-28”的数据集,依托于LeRobot框架构建,专注于单臂Franka机器人的精细操控,虽仅包含单条示范轨迹与243帧数据,却蕴藏着前沿研究的核心脉络。该数据集通过深度相机获取的164个三维空间点云坐标与机械臂七自由度关节角度、末端执行器位姿及夹爪状态等异构传感信息深度融合,为模仿学习与行为克隆任务提供了高保真的多模态观测空间。当前,研究趋势聚焦于如何从此类小样本、高维度的数据中高效提取通用操作先验,结合扩散模型或基于Transformer的策略架构,推动机器人从单一技能演示迈向泛化性更强的自主作业。这一数据集的发布,不仅呼应了学界对低成本、标准化机器人数据采集规范的迫切需求,更在无形中为构建可迁移的机器人操作知识库奠定了实证基础,其对强化学习预训练与少样本适应技术发展的启示意义不容小觑。
以上内容由遇见数据集搜集并总结生成



