遇见数据集

office-rover-2task-raw

收藏
Hugging Face2026-09-08 更新2026-09-04 收录
官方服务:

资源简介:

一个用于单臂SO-101机器人的LeRobot v2.1数据集,任务为“拾取红色/绿色立方体并放入盒子”,两个立方体在桌上。包含两个语言条件任务(指令为俄语)。专为在单张24 GB RTX 4090上微调NVIDIA Isaac GR00T N1.7而构建。数据集包含500个演示片段,共69000帧(30 fps),两个摄像头(前视和腕部,分辨率240×320),状态/动作为6个关节(5个关节+夹爪),绝对目标。任务在MuJoCo模拟器中生成,使用官方SO-101模型(来自mujoco_menagerie),演示由脚本操作员(逆运动学)生成,非人类遥操作。立方体位置在12×30 cm区域内随机,成功判断基于物理仿真(指定立方体在盒子中,另一立方体未被触碰)。策略看不到立方体坐标。未模拟真实光照、眩光、摩擦和伺服回差,仿真到现实的差距未测量。可直接通过LeRobotDataset("VShirokun/office-rover-2task-raw")加载。许可协议:CC BY 4.0。

A LeRobot v2.1 dataset for a single-arm SO-101 robot with the task pick up the red/green cube and put it in the box, with two cubes on the table. Contains two language-conditioned tasks (instructions in Russian). Built specifically for fine-tuning NVIDIA Isaac GR00T N1.7 on a single 24 GB RTX 4090. The dataset includes 500 demonstration episodes, totaling 69,000 frames (30 fps), two cameras (front view and wrist, resolution 240×320), state/action for 6 joints (5 joints + gripper), absolute targets. The task is generated in MuJoCo simulator using the official SO-101 model (from mujoco_menagerie), demonstrations are generated by a scripted operator (inverse kinematics), not human teleoperation. Cube positions are randomized within a 12×30 cm area, success is determined based on physics simulation (specified cube in the box, the other cube untouched). The policy does not see the cube coordinates. Real-world lighting, glare, friction, and servo backlash are not simulated; the sim-to-real gap is not measured. Can be loaded directly via LeRobotDataset("VShirokun/office-rover-2task-raw"). License: CC BY 4.0.

提供机构:
VShirokun
创建时间:
2026-09-04
原始信息汇总

数据集概述:office-rover-2task-raw

该数据集是一个用于机器人操作任务(Robotics)的LeRobot v2.1格式数据集,专为单只SO-101机械臂设计,目标任务是语言条件下的抓取与放置操作。

核心任务

  • 任务指令(俄语):“拿起红色/绿色方块并放入盒子中”
  • 两个语言条件任务,两个方块均置于桌面上
  • 数据集专为在单张24 GB RTX 4090上微调NVIDIA Isaac GR00T N1.7模型而构建(相关配方及评估细节见GitHub仓库)

数据规模与技术规格

项目 规格
演示片段数(episodes) 500
总帧数 69,000(30 fps)
相机 front(前视)、wrist(腕部),分辨率240×320
状态/动作 6关节(5个旋转关节+1个夹爪),绝对目标控制
任务数量 2

数据收集方式(重要声明)

  • 仅仿真环境:基于MuJoCo物理引擎,使用mujoco_menagerie中的官方SO-101模型
  • 演示来源:由脚本化操作员(逆向运动学)生成,并非人工遥操作
  • 方块位置在12×30 cm区域内随机分布,任务成功由物理引擎判定(目标方块在盒内且另一方块未被触碰)
  • 策略模型在训练和评估中无法获取方块坐标信息
  • 局限性:未模拟真实光照、眩光、摩擦及伺服回程差,仿真到现实的迁移差距未经验证

使用与许可

  • 加载方式:LeRobotDataset("VShirokun/office-rover-2task-raw")
  • 数据由LeRobot v3.0格式转换而来(使用上游scripts/lerobot_conversion/convert_v3_to_v2.py脚本)
  • 许可证:CC BY 4.0
  • 引用要求:请引用上述GitHub仓库

模型评估参考(原始数据)

  • 在不做额外工程处理的前提下直接训练,GR00T N1.7达到386/600 = 64.33%(置信区间60.4–68.2)的成功率
  • 对比参照:一个450M参数量的VLA模型在该数据上得分仅为0–6%
二维码
社区交流群
二维码
科研交流群
商业服务