five

RoboCOIN/AI2_Alphabot_2_place_pen

收藏
Hugging Face2026-06-08 更新2026-06-14 收录
下载链接:
https://hf-mirror.com/datasets/RoboCOIN/AI2_Alphabot_2_place_pen
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集名为AI2_Alphabot_2_place_pen,是一个机器人操作数据集,专注于将笔放入笔筒中的任务。它基于LeRobot格式并完全兼容LeRobot框架。数据集包含527个episodes,总计666,384帧,大小为13.18 GB,帧率为30 FPS。使用AI2_Alphabot_2机器人,配备双指末端执行器,并通过VR控制器进行遥操作。传感器包括四个RGB摄像头:前胸摄像头、前头摄像头、左腕摄像头和右腕摄像头,分辨率均为640x480。场景设置为教育->学校环境,涉及物体包括白色笔筒、黑色圆珠笔和黑色钢笔。数据包含状态信息(34维,如关节角度、末端执行器位姿)和动作信息(34维),并提供丰富的注释,如末端执行器加速度、方向、速度等。数据集由北京智源人工智能研究院的RoboCOIN团队贡献,用于机器人学习和操作研究。

This dataset, named AI2_Alphabot_2_place_pen, is a robotics manipulation dataset focused on the task of placing the pen into the pen holder. It uses an extended format based on LeRobot and is fully compatible with LeRobot. The dataset contains 527 episodes, totaling 666,384 frames, with a size of 13.18 GB and a frame rate of 30 FPS. It employs the AI2_Alphabot_2 robot with a two-finger end-effector and VR controller teleoperation. Sensors include four RGB cameras: cam_front_chest_rgb, cam_front_head_rgb, cam_left_wrist_rgb, and cam_right_wrist_rgb, all with a resolution of 640x480. The scene is set in an Education->School environment, with objects including a white pen holder, a black ballpoint pen, and a black fountain pen. The data includes state information (34-dimensional, such as joint angles and end-effector poses) and action information (34-dimensional), along with rich annotations like end-effector acceleration magnitude, direction, and velocity. Contributed by the RoboCOIN Team at Beijing Academy of Artificial Intelligence (BAAI), it is designed for robot learning and manipulation research.
提供机构:
RoboCOIN
搜集汇总
数据集介绍
main_image_url
构建方式
在机器人学习领域,精确的操作技能习得依赖于高质量示范数据的采集。AI2_Alphabot_2_place_pen数据集由RoboCOIN团队在北京人工智能研究院构建,专注于教育场景下的笔类物品放置任务。该数据集采用基于LeRobot标准的扩展格式,通过VR控制器遥操作AI2_Alphabot_2双机械臂机器人,在真实世界环境中采集了527个示范回合,共计666,384帧图像和状态动作序列。数据以parquet文件存储于单一数据块中,并同步保存四组RGB摄像头视频流,支持完备的示教学习流程。
使用方法
使用者可借助LeRobot框架快速加载与解析该数据集。数据文件遵循data/chunk-{id}/episode_{id}.parquet的路径模式,视频文件则按摄像头视角组织于对应子目录中。通过读取parquet文件中的observation.state和action数组,可获取完整的34维机器人状态与动作指令。同时,位于annotations目录下的JSONL注释文件支持按需接入逐帧语义标签。数据集已划分为单一训练集,覆盖全部527个回合,便于直接用于行为克隆、逆强化学习等算法的模型训练与评估。
背景与挑战
背景概述
精细操作技能赋予机器人执行复杂任务的能力,是衡量智能体感知与交互水平的关键维度。AI2_Alphabot_2_place_pen数据集由北京人工智能研究院(BAAI)RoboCOIN团队于2025年创建,专注于双机械臂协同下的笔具放置这一典型精细操作任务。该数据集依托AI2_Alphabot_2机器人平台,通过VR控制器采集了527条示范轨迹,累计超过66万帧画面,涵盖四视角RGB视频及34维状态-动作空间,为研究从示范中学习精细操作提供了高质量、结构化的资源。作为RoboCOIN项目的重要组成部分,该数据集旨在推动可泛化操作技能学习,其发布显著丰富了机器人学习领域的数据生态,尤其在双臂协同与精确放置任务的研究范式中具有里程碑意义。
当前挑战
该数据集的核心挑战在于双机械臂协同下完成笔具至笔筒的精确放置操作,其复杂性远超单臂抓取任务,需解决双臂运动协调、末端执行器姿态控制以及目标空间定位的精准性问题。同时,真实世界环境中的光照变化、物体表面反光特性及周围干扰物进一步增加了视觉感知与动作执行的难度。数据集构建过程中,采集高质量示范轨迹亦面临严峻考验:VR遥操作虽能捕捉细腻的人类操作策略,但操作者需克服延迟与力反馈缺失以实现平滑运动,不同操作者的示教风格差异也带来了数据一致性的问题。此外,对34维状态空间进行高频同步标注,以及妥善处理各传感器采集频率不统一引发的时空对齐偏差,均为数据集构建过程中的技术壁垒。
常用场景
经典使用场景
在机器人学习与操作领域,AI2_Alphabot_2_place_pen数据集为双机械臂协同精细操作研究提供了宝贵的真实世界示范数据。其核心任务是将笔精准放入笔筒,涵盖拾取、传递和放置等原子动作,涉及物体感知、夹爪控制与双臂协调。该数据集的经典用途在于训练基于视觉的运动策略,尤其是利用多视角RGB视频(胸部、头部及左右腕部摄像头)作为观测输入,通过模仿学习或强化学习方法,驱动AI2_Alphabot_2机器人完成高精度插拔式装配任务。它特别适合验证策略在真实场景下对物体位姿变化和夹取力度的鲁棒性。
解决学术问题
该数据集直面双机械臂在非结构化环境中的协同操作难题,解决了真实世界数据稀缺对双臂模仿学习发展的制约。它推动了对“观察-动作”映射关系的深层次研究,特别是在多模态感知融合(视觉、关节状态与夹爪开度)与精细动作序列生成方面的学术探索。数据集附带的丰富语义标注(如末端执行器方向、速度、夹爪活动模式)为拆解复杂操作层级、研究动作基元与高层次任务规划之间的关系提供了可靠基准。其研究与发布标志着开源双臂数据集在场景真实性和标注丰富度上的重要进步,加速了可复现的双臂精细操作算法评估体系的建立。
实际应用
AI2_Alphabot_2_place_pen数据集最直接的应用场景指向教育机器人,特别是学校环境中的演示与教学。实际部署中,该数据集可用于训练机器人完成教室桌面整理等任务,例如自动拾取散落的文具并归位,通过视觉引导实现目标识别与抓取。此外,在轻工业装配线,机器人可借助此类数据学习小零件(如笔类物件)的精确插入工序。家庭服务机器人也能从中受益,学习完成诸如将筷子放入筷笼、笔放入笔筒等日常生活的精准放置动作。它推动机器人从重复预设程序向具备环境适应能力的自主操作转变。
数据集最近研究
最新研究方向
该数据集聚焦于具身智能领域中的双臂协作精细操作任务,以“将笔放入笔筒”这一典型教育场景为切入点,为机器人学习提供了高保真的多模态示范数据。当前,该数据集正被广泛用于训练和评估基于模仿学习与强化学习的双臂协同操作策略,尤其在与视觉-语言模型结合的机器人指令跟随研究中展现出重要价值。其丰富的传感器配置、精细的末端执行器控制参数以及标准化的LeRobot格式,极大地降低了复现门槛,推动了RoboCION等开源项目的进展。这一数据资产的涌现,标志着机器人操作研究正从单一机械臂向复杂双臂协同、从实验室模拟向真实教学环境迁移的范式转变。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务