RoboCOIN Dataset
收藏RoboCOIN数据集概述
数据集基本信息
- 数据集名称:RoboCOIN
- 官方工具包:本项目是[RoboCOIN数据集]的官方配套工具包。
- 技术报告:可在ArXiv获取。
- 项目网站:https://flagopen.github.io/RoboCOIN/
- 数据管理页面:https://flagopen.github.io/RoboCOIN-DataManager/
- 数据下载量:截至2025年12月9日,在HuggingFace和ModelScope上的总下载量已达300,000次。
核心特性
- 数据集管理:提供数据集的检索、下载和基于
DataLoader的加载,完全支持子任务、场景和运动描述等丰富的元数据注释。 - 统一的机器人控制接口:支持集成多种机器人平台,包括基于SDK的控制(如Piper、Realman)和通用的基于ROS/MoveIt的控制。
- 标准化单位转换:内置跨平台单位处理工具(例如度与弧度转换)。
- 可视化工具:提供2D/3D轨迹绘图和同步相机图像渲染。
- 策略推理与部署:为LeRobot策略和OpenPI策略提供开箱即用的推理流程,支持从训练好的模型直接控制机器人。
数据集格式与兼容性
- 基础框架:基于LeRobot仓库构建。
- 数据格式:完全兼容LeRobot的数据格式。
- 元数据增强:在LeRobot格式基础上,增加了对子任务、场景描述和运动描述等丰富元数据的支持。
- 版本兼容性:
- 当前支持LeRobot v2.1数据格式。
- 对v3.0数据格式的支持即将推出。
- 代码库:当前基于LeRobot v0.3.4,未来版本将演变为完全兼容的LeRobot扩展插件。
数据特征说明
observation.state / action
这些特征代表从机器人手臂(从端/主端)收集的数据。在没有机器人动作数据的情况下,动作由observation.state序列推导得出。标准化字段如下:
| 特征 | 单位 | 描述 |
|---|---|---|
{dir}_arm_joint_{num}_rad |
rad | 从收集数据转换而来;代表手臂关节角度(从端/主端)。 |
{dir}_hand_joint_{num}_rad |
rad | 从收集数据转换而来;代表手部关节角度。 |
{dir}_gripper_open_scale |
- | 取值范围[0, 1];0表示完全闭合,1表示完全打开;从收集数据转换而来。 |
{dir}_eef_pos_{axis} |
m | 从机器人SDK获取的末端执行器位置。 |
{dir}_eef_rot_{axis} |
rad | 从机器人SDK获取的末端执行器旋转(欧拉角)。 |
eef_sim_pose_state / eef_sim_pose_action
这些特征代表模拟的末端执行器数据。由于不同机器人SDK在observation.state / action数据中的坐标系定义不一致,我们采用基于模拟的方法来获取每个机器人在统一坐标系(x轴向前 / y轴向左 / z轴向上,原点位于机器人基座或其双脚中心)中表达的末端执行器位姿。这些模拟的末端执行器位姿由特征eef_sim_pose_state / eef_sim_pose_action表示。
注意:
{dir}是一个占位符,代表left或right。
安装与使用
安装
bash pip install robocoin
数据集发现与下载
-
下载命令示例: bash
从HuggingFace下载
robocoin-download --hub huggingface --ds_lists Cobot_Magic_move_the_bread R1_Lite_open_and_close_microwave_oven
从ModelScope下载
robocoin-download --hub modelscope --ds_lists Cobot_Magic_move_the_bread R1_Lite_open_and_close_microwave_oven
-
默认下载路径:
~/.cache/huggingface/lerobot/,该路径将作为LerobotDataset的默认目录。
加载数据集
python import torch from lerobot.datasets.lerobot_dataset import LeRobotDataset
dataset = LeRobotDataset("RoboCOIN/R1_Lite_open_and_close_microwave_oven") dataloader = torch.utils.data.DataLoader( dataset, num_workers=8, batch_size=32, )




