遇见数据集

maysmile/icra_0526_bottle_sub

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

资源简介:

该数据集由LeRobot创建,是一个用于机器人学任务的数据集,包含201个episodes,总计68555帧数据。数据以parquet格式存储,视频以mp4格式存储。数据集的特征包括:动作(action),表示机器人左右臂和夹爪的控制信号;观测状态(observation.state),表示机器人左右臂和夹爪的当前状态;以及来自前、右、左三个视角的图像观测(observation.images.front/right/left),图像分辨率为480x640或640x480,帧率为15fps,编码为av1。此外,还包括时间戳、帧索引、episode索引、索引和任务索引等元数据。机器人类型为rby1,数据集总大小为300MB(数据文件100MB,视频文件200MB),适用于机器人控制、视觉导航等研究。

This dataset was created using LeRobot and is designed for robotics tasks. It contains 201 episodes with a total of 68,555 frames. The data is stored in parquet format, and videos are stored in mp4 format. The dataset features include: action, representing control signals for the robots right and left arms and grippers; observation.state, representing the current state of the robots arms and grippers; and observation.images.front/right/left, providing image observations from front, right, and left perspectives with resolutions of 480x640 or 640x480, a frame rate of 15fps, and av1 encoding. Additionally, metadata such as timestamp, frame_index, episode_index, index, and task_index are included. The robot type is rby1, and the total dataset size is 300MB (100MB for data files and 200MB for video files). It is suitable for research in robot control, visual navigation, and related fields.

提供机构:
maysmile
搜集汇总
数据集介绍
maysmile/icra_0526_bottle_sub 数据集图片
构建方式
该数据集借助LeRobot框架构建,专注于机器人操作任务。数据采集过程中,研究者精心设计了针对瓶子抓取与放置的子任务场景,通过机器人本体传感器记录高精度的机器人状态与动作序列。数据集的结构清晰且规范,包含多模态信息:16维的动作指令(涵盖双臂7自由度关节位置及夹爪控制)、对应的观测状态向量,以及来自前置、左侧和右侧三个视角的高清彩色视频流。所有数据均以15帧/秒的帧率同步采集,并划分为201个独立的情节(episode),总计包含68555帧连续样本,确保了时间序列上的连贯性与完整性。
使用方法
该数据集无缝集成于LeRobot生态系统,用户可通过HuggingFace上的可视化工具直观浏览样本片段。对于开发者,推荐使用LeRobot库加载数据:通过`dataset = load_dataset('maysmile/icra_0526_bottle_sub', use_videos=True)`一键获取包含视频与状态数据的多模态字典。数据集遵循Apache-2.0许可,适用于学术研究。在训练机器人策略时,可将16维动作与状态向量直接映射至rby1型机器人的控制接口,并结合三视角图像进行端到端的模仿学习或强化学习研究。数据已按1000帧的大小分块(chunk)存储,高效支持大规模训练流水线。
背景与挑战
背景概述
该数据集名为icra_0526_bottle_sub,创建于近年来机器人学习领域快速发展的背景下,由maysmile团队基于LeRobot框架构建,旨在为机械臂操作任务提供标准化训练样本。数据集聚焦于瓶子抓取与放置的精细操作场景,通过记录rby1型双机械臂的关节角度、夹爪状态及多视角视觉信息(正面、左侧、右侧摄像头),为模仿学习与强化学习算法提供高保真度的多模态输入。其核心研究问题在于如何利用少量示教数据泛化复杂操作策略。数据集包含201个示范片段与约6.9万帧数据,以15帧/秒同步动作-状态-图像序列,为机器人操作领域的基础研究提供了可复现的基准资源。
当前挑战
该数据集所解决的领域挑战在于机器人操作任务中的精准模仿与泛化能力不足,尤其在瓶类物品的拾取、旋转与放置等非结构化操作中,传统控制方法难以适应细微姿态变化与物体形状差异。构建过程中面临多项技术难题:首先,双机械臂14自由度协同运动需保证动作标注的时空一致性,数据采集时需消除传感器延迟导致的异步问题;其次,多视角视频数据(AV1编码、不同分辨率)的压缩与标准化存储需平衡画质与存储效率;最后,单任务设定下如何通过200MB的小样本数据规模(含100MB结构化动作/状态数据与200MB视频)设计高效的数据增强策略,以防止过拟合并提升跨物体泛化性。
常用场景
经典使用场景
在机器人操作与模仿学习的研究疆域中,数据集是推动智能体从感知走向行动的基石。icra_0526_bottle_sub数据集聚焦于双机械臂协同操作瓶装物体的精细任务,通过LeRobot框架采集并结构化存储,包含前视、左右多视角视觉观测、16维关节与夹爪状态信息,以及对应的动作序列。该数据集最经典的使用场景是作为模仿学习(Imitation Learning)与行为克隆(Behavioral Cloning)的训练基准,研究者可基于多模态观测序列,训练机器人从专家演示中习得拧瓶盖、抓取与放置等操作策略,尤其适用于端到端视觉-运动控制模型的研发与评估。
解决学术问题
该数据集精准回应了机器人领域两个核心学术难题:一是高维连续动作空间下的策略学习如何在有限演示数据中保持良好的泛化性,二是多指手臂协同操作时的运动与力约束建模问题。通过提供201个高质量演示片段,它使得研究者能够系统性地探索条件模仿学习、潜变量建模以及数据增强技术对操作鲁棒性的影响。其意义在于,将瓶装物体这类非刚性且需精细力控的典型任务从单一场景扩展到可复现的标准化评估,推动了关于接触模型、对称性利用以及多视角融合等基础问题的定量研究。
实际应用
在现实工业与家庭服务场景中,瓶装物体的操作无处不在——从饮料灌装线到老年辅助机器人开瓶。icra_0526_bottle_sub数据集为这些场景提供了可直接迁移的算法验证平台。基于该数据集训练的视觉伺服策略可部署于RBY1型人形机器人,使其在仓库或厨房环境中自主完成瓶体抓取、拧盖和分拣。此外,其多视角图像与关节状态配对的数据结构,直接支撑了力控远程操作系统的仿真器替代方案,降低了真实机器人调试成本,加速了零售自动化与医疗辅助设备中灵巧操作的落地进程。
数据集最近研究
最新研究方向
在机器人学习领域,该数据集以其精细化的双臂协作任务设定,成为模仿学习与行为克隆前沿探索的重要基石。依托LeRobot框架构建的icra_0526_bottle_sub,聚焦于机器人对瓶子类物体的精准操作,其多视角视觉输入与16维动作空间的设计,为研究复杂灵巧操作中的状态表征与策略泛化提供了高质量数据支撑。当前,该数据集正被广泛应用于推动具身智能体在非结构化环境中执行精细抓取与装配任务的能力演进,尤其与热门的人形机器人协同操作、少样本模仿学习等研究方向紧密关联,为打破机器人技能迁移瓶颈、实现通用操作智能贡献了关键实证与基准。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务