遇见数据集

ecappiell/weighted_boxes_experiment_20260529_091235

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

资源简介:

该数据集是使用LeRobot创建的机器人学数据集,专注于机器人控制与感知任务。数据集包含动作和观测数据,其中动作由6个浮点数值组成,对应机器人关节(肩部平移、肩部升降、肘部弯曲、腕部弯曲、腕部旋转和抓取器)的位置控制。观测数据包括状态和图像:状态特征有111个维度,涵盖所有关节的位置、原始电流、负载、硬件速度、速度、原始扭矩、带符号的电流和扭矩、滤波速度、估计外部扭矩以及模型计算的多种扭矩(电机、重力、摩擦、科里奥利、惯性和模型扭矩),还包括TCP(工具中心点)的力和力矩,以及HPI抓取器的外部扭矩、位置和速度;图像特征提供两个视角的视频(顶部和角落),分辨率均为480x640,3通道,帧率30fps,使用AV1编解码器。数据集总共有5个剧集、3498帧和1个任务,数据以parquet文件存储,视频以mp4文件存储,适用于机器人学习和强化学习应用。

This dataset is a robotics dataset created using LeRobot, focusing on robot control and perception tasks. It includes action and observation data: actions consist of 6 float values corresponding to position control of robot joints (shoulder pan, shoulder lift, elbow flex, wrist flex, wrist roll, and gripper). Observations comprise state and images: state features have 111 dimensions, covering joint positions, raw currents, loads, hardware velocities, velocities, raw torques, signed currents and torques, filtered velocities, estimated external torques, and model-calculated torques (motor, gravity, friction, coriolis, inertial, and model torques) for all joints, as well as TCP (Tool Center Point) forces and torques, and HPI gripper external torque, position, and velocity; image features provide video from two perspectives (top and corner), each with a resolution of 480x640, 3 channels, 30fps frame rate, using AV1 codec. The dataset totals 5 episodes, 3498 frames, and 1 task, with data stored in parquet files and videos in mp4 files, suitable for robotics learning and reinforcement learning applications.

提供机构:
ecappiell
搜集汇总
数据集介绍
ecappiell/weighted_boxes_experiment_20260529_091235 数据集图片
构建方式
该数据集基于LeRobot框架构建,专为机器人操作任务设计。数据采集自SO-100系列机器人,以30帧每秒的频率记录高保真操作序列。数据集由5个完整片段组成,涵盖3498帧时序数据,每个片段均包含机器人从初始状态到任务完成的完整轨迹。数据以Parquet格式存储结构化信息,同时视频数据采用AV1编码压缩,存储于独立的MP4文件中,实现了高效的数据管理与存取。为确保数据完整性,每个片段均记录了时间戳、帧索引、片段索引及任务索引等元信息。
使用方法
数据集通过HuggingFace的LeRobot库实现标准化加载与使用。用户可借助LeRobot框架的dataset API,一键获取标准化的'action'、'observation.state'及'observation.images'等数据字段。数据已预先划分为训练集,可直接用于模仿学习与强化学习算法的训练与评估。对于视觉-运动策略的开发,建议结合视频流与状态数据进行多模态建模。完整的可视化与调试支持可通过LeRobot在线平台实现,便于研究者在实际部署前直观评估数据质量。
背景与挑战
背景概述
本数据集“weighted_boxes_experiment_20260529_091235”由Hugging Face社区成员ecappiell基于LeRobot框架创建,旨在推动机器人操作任务的模仿学习研究。该数据集聚焦于机械臂在执行带有重量变化的箱体操作实验中的精细控制,核心研究问题在于如何利用多模态观测数据(包括视觉图像与关节状态、扭矩等本体感知信息)训练机器人实现鲁棒的操作技能。数据集包含5个回合、约3498帧的高频(30fps)记录,涵盖了6自由度机械臂的完整运动与力觉反馈。其发布为机器人领域提供了一个标准化的小规模但高维度示范数据集,有助于验证模仿学习算法在真实物理交互环境中的泛化能力与数据效率。
当前挑战
该数据集所解决的领域挑战在于机器人对非结构化物体的灵巧操作,尤其是面对负载变化时维持精准控制与自适应策略的难题。具体而言,构建过程中面临多重挑战:首先,需要同步采集高分辨率视觉(640×480像素)与高达111维的关节本体感知数据,确保时间对齐与数据一致性;其次,仅5个回合(3498帧)的小样本训练数据对模仿学习算法提出了严苛要求,需在数据稀缺下避免过拟合并提取通用操作模式;此外,真实物理环境中传感器噪声、扭矩反馈的非线性以及机械臂动力学模型的复杂性,增加了从示范中提炼稳定策略的难度。
常用场景
经典使用场景
在机器人操作与模仿学习领域,该数据集专为训练机械臂完成精细抓取与放置任务而设计,尤其聚焦于不同重量与尺寸的箱子操作场景。通过采集Franka Emika Panda机械臂在6自由度关节空间下的动作指令、高维状态观测(包括关节位置、速度、电流、力矩及扭矩估计)以及多视角视觉输入(顶部与角落摄像头),研究者可构建从感知到动作的端到端策略模型。经典应用方向包括基于行为克隆的模仿学习、逆强化学习以及离线强化学习,用以复现专家演示中的复杂操控技能。
解决学术问题
该数据集着力破解机器人操作中两大核心学术难题:变负载条件下的力位混合控制与多模态感知融合。传统研究常受限于恒定负载假设,而本数据集中详细的关节力矩、外部扭矩估计以及工具中心点(TCP)力/力矩信息,为探索负载变化下的自适应控制策略提供了关键线索。同时,同步记录的高清视觉帧与多维状态数据,促使研究者量化视觉-本体感觉耦合在精密操作中的作用,推动了关于动态环境鲁棒性和泛化能力的理论突破。
实际应用
在实际产业场景中,该数据集为生产线自动化包装、物流分拣以及精密仪器装配等任务提供了标准化验证平台。特别是对于涉及不同重量箱体搬运的弹性制造单元,基于数据集训练的策略可直接迁移至真实机械臂,实现视觉引导下的自适应抓取与柔性放置。此外,数据集中的力矩和电流数据有助于工业机器人实现碰撞检测与安全交互,为协作机器人部署于人类共享工作空间奠定了数据基础。
数据集最近研究
最新研究方向
在机器人学习与模仿学习的前沿领域,该数据集依托LeRobot框架采集,聚焦于六自由度机械臂的精细化操作任务。其观测空间包含了关节位置、速度、电流、扭矩及外部力矩估计等111维高保真状态信息,并配以双视角高清视觉流,为研究基于多模态融合的机器人技能习得提供了丰富的物理与视觉特征。当前,结合大规模预训练模型与端到端模仿学习的热点事件,此类细粒度、包含力矩与接触力信息的真实操控数据集,正支撑着从动作原语到动态环境适应性的关键突破,推动机器人操作泛化能力迈向更高阶的智能层级。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务