five

TrossenRoboticsCommunity/widowx_ai_orange_block_transfer

收藏
Hugging Face2026-06-10 更新2026-06-14 收录
下载链接:
https://hf-mirror.com/datasets/TrossenRoboticsCommunity/widowx_ai_orange_block_transfer
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集是使用LeRobot平台创建的,专为机器人学任务设计。数据集包含20个训练片段,总帧数为13180,帧率为30fps。数据以parquet格式存储,视频以mp4格式存储。数据集特征包括动作数据(14维关节位置,对应左右机械臂的关节和滑轨位置)、观察状态(同样14维关节位置),以及来自四个摄像头的图像观察:低角度摄像头、高角度摄像头、右腕摄像头和左腕摄像头,所有图像分辨率为480x640,3通道,使用AV1编解码器。此外,还包括时间戳、帧索引、片段索引、任务索引等元数据。机器人类型为双机械臂WidowX AI跟随机器人,适用于机器人控制、视觉感知和强化学习等研究。

This dataset was created using the LeRobot platform, specifically designed for robotics-related tasks. It contains 20 training episodes, with a total of 13,180 frames at a frame rate of 30 fps. The structured data of the dataset is stored in Parquet format, while the video files are stored in MP4 format. The dataset includes the following features: action data (14-dimensional joint positions, corresponding to the joints of the left and right robotic arms and the positions of the linear slider), observation states (also 14-dimensional joint positions), and visual observations from four cameras: a low-angle camera, a high-angle camera, a right-wrist camera, and a left-wrist camera. All images have a resolution of 480×640, 3 color channels, and use the AV1 codec. Additionally, it includes metadata such as timestamps, frame indices, episode indices, and task indices. The robotic platform involved is a dual-arm WidowX AI following robot, which is suitable for research in robot control, visual perception, reinforcement learning, and other related fields.
搜集汇总
数据集介绍
main_image_url
构建方式
在机器人学习领域,模仿学习依赖高质量的人机交互数据来训练策略模型。该数据集基于LeRobot框架构建,通过双WidowX AI机器人主从遥操作方式采集,专注于橙色方块转移这一单一任务。数据集包含20个完整演示回合,共计13180帧时序数据,以30帧每秒的采样频率记录了机器人双臂的14维关节动作与状态信息。数据以Parquet格式存储结构化时序数据,同时通过AV1编码的MP4视频保存四个视角(低位、高位、左腕、右腕)的640×480分辨率视觉观测,确保动作状态与视觉信息在时间轴上的精确对齐。
特点
该数据集的核心特色在于其多维度的精细数据表征。动作与状态空间均采用14维连续值描述双臂六自由度关节及基座姿态,视觉信息涵盖四个同步摄像机视角,为多模态融合研究提供基础。数据按回合组织,每回合包含完整的动作序列、状态轨迹与多视角视频流,时间戳和帧索引的加入便于时序建模。此外,数据集明确划分训练集(100%用于训练),并提供了100MB的结构化数据与200MB的视频文件,体现了对存储效率与数据完整性的平衡。
使用方法
使用者可通过LeRobot库的API加载该数据集,直接利用其预定义的数据分割与特征结构进行模型训练。数据集的action和observation.state字段提供了14维关节参数,可配合四路视觉观测(cam_low、cam_high、cam_right_wrist、cam_left_wrist)构建模仿学习或强化学习算法。Parquet格式支持高效的列式读取,而AV1编码的视频有效压缩视觉数据,便于在机器人策略网络中进行端到端训练。该数据集特别适用于双臂协调操作任务的研究,也可作为迁移学习或数据增强的基准资源。
背景与挑战
背景概述
该数据集由Trossen Robotics社区基于LeRobot框架创建,聚焦于双机械臂协同操作任务,具体为WidowX AI机器人执行橙色方块转移的精细化操作。在机器人学习领域,模仿学习与行为克隆技术对高质量、多模态的示教数据需求日益增长,尤其需要涵盖精确关节角度、多视角视觉信息与时间序列对齐的复合型数据集。该数据集通过记录14维关节角度状态、4个摄像头视角(低机位、高机位及左右腕部)的高清视频流及对应动作序列,为双臂协同操作研究提供了标准化数据支撑。作为开源社区贡献的成果,其Apache-2.0许可协议促进了机器人操作任务的复现与基准测试,对推动智能机器人精细操作能力的发展具有重要参考价值。
当前挑战
在领域问题层面,双臂机器人协同操作面临运动学冗余、任务空间约束与动态平衡等核心挑战,而橙色方块转移任务要求机械臂在非结构化环境中实现精准抓取、稳定传递与平滑放置,这对动作规划的实时性与鲁棒性提出极高要求。在数据集构建过程中,挑战体现为:多摄像头同步采集难度大,需确保4路640×480分辨率视频流与30Hz关节运动数据保持微秒级时间对齐;20个示范片段虽能初步覆盖任务空间,但多样性不足,难以泛化至不同初始方块位姿与光照条件;此外,数据集仅包含单一任务类型,缺乏对失败案例的标注与补偿策略的记录,限制了基于该数据训练的模型在复杂场景下的自适应能力。
常用场景
经典使用场景
widowx_ai_orange_block_transfer 数据集是面向双臂协作机器人操控任务的经典资源,依托 Trossen Robotics 的 WidowX AI 平台构建。该数据集由 20 个演示片段、共 13180 帧视觉与状态序列组成,涵盖左右臂共计 14 个自由度的关节动作以及四路摄像头(低位、高位、左腕、右腕)的同步 RGB 视频。在机器人学习领域,它被广泛用于模仿学习(Imitation Learning)和行为克隆(Behavioral Cloning)的基准训练,研究者利用其中细粒度的动作标签与多视角视觉观测,训练模型从人类演示中习得橙色方块抓取、转移与放置的灵巧操作策略。数据集以 30 FPS 的帧率提供密集的时序信息,支持对连续动作空间的精确建模,成为验证双臂协同控制算法效果的标准测试平台。
解决学术问题
该数据集着力于解决双臂机器人在精密物体操作中的两个核心学术难题:多机械臂间的协调规划与视觉引导下的精准抓取。传统单臂研究往往忽视双臂在空间中的位姿耦合与冲突避免,而 widowx_ai_orange_block_transfer 通过提供同步的左右臂关节状态与多视角图像,使得学术界能够探索基于视觉的双臂协同策略,例如联合注意力机制与共享表征的学习。此外,数据集中的橙色方块转移任务对末端执行器的定位精度提出高要求,推动了从像素到关节力矩的端到端模型研究,提升了在不确定性环境中(如光照变化、物体轻微偏移)的操作鲁棒性。这些研究对于构建通用且灵活的机器人操作智能具有奠基性意义,缩小了模拟环境与真实世界部署之间的鸿沟。
衍生相关工作
widowx_ai_orange_block_transfer 的出现催生了一系列围绕双臂模仿学习与视觉运动策略的衍生工作。研究者借助该数据集验证了多种先进算法,例如扩散策略(Diffusion Policy)用于生成平滑的多模态动作轨迹,以及基于 Transformer 的架构如 ACT(Action Chunking Transformer)从时序视觉中预测连续运动片段。此外,数据集也被用作评估数据增强技术(如视角随机化与域随机化)在现实机器人学习中的有效性基准。部分工作还尝试结合该数据集与仿真环境进行策略迁移,如利用域自适应方法弥合合成数据与真实物理世界的差异。这些衍生贡献不仅推动了双臂机器人学习的理论深化,还为后续大规模多任务数据集的构建提供了方法论参考,彰显了该数据集在机器人学习社区中的桥梁作用。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务