PRTS-PostTrain-Data
收藏PRTS Post-Training Data 数据集概述
数据集基本信息
- 数据集名称:PRTS Post-Training Data
- 发布机构/项目:PRTS (A Primitive Reasoning and Tasking System via Contrastive Representations)
- 维护者:TeleEmbodied AI Team
- 许可协议:MIT License
- 数据规模:约 50GB
数据集描述
该数据集是一个高质量机器人操作数据集,专为学习原始技能和长时程任务执行而设计。数据采用标准 LeRobot 格式进行整理。
核心特点
- 操作类型:涵盖双臂协调与单臂操作。
- 任务复杂度:包含精细操作与长距离操作。
- 任务规划:支持长时程任务。
- 数据来源:数据采集自真实机器人系统,覆盖家庭、办公室和工业环境。
数据内容
每条轨迹记录了详细的控制信号,包括:
- 位姿
- 关节状态
- 夹爪动作 这些数据可用于学习原始技能、动作表示和任务级策略。
主要用途
直接用途
- 学习双臂与单臂操作策略。
- 训练原始级动作表示。
- 长时程任务学习与分解。
- 精细操作技能学习。
- 在真实场景中为机器人操作模型提供基准测试。
数据结构
数据集遵循标准 LeRobot 格式。
任务结构
每个任务包含结构化的轨迹,具有逐帧的机器人状态:
task_name:任务名称(例如:"Pick and Place Object")。task_id:唯一任务标识符。task_description:任务描述。frames:帧序列,每帧包含:t:时间戳。pose:6D末端执行器位姿(双臂或单臂)。joints:关节角度。gripper:夹爪状态(开/闭)。
数据示例
json { "task_name": "Pick and Place Object", "task_id": "001", "task_description": "Pick up a cup and place it on the table", "frames": [ { "t": 0.0, "pose": [0.5, 0.5, 0.2, 0, 0, 0, 1], "joints": [0.3, -0.5, 0.8, 0.1, -0.2, 0.9, 0.2, -0.4], "gripper": [1, 0] } ] }
引用信息
若使用此数据集,请引用:
@article{zhang2026prts, title={PRTS: A Primitive Reasoning and Tasking System via Contrastive Representations}, author={Zhang, Yang and Zhao, Jiangyuan and Fan, Chenyou and Yan, Fangzheng and Li, Tian and Wu, Xuaner and Weng, Qizhen and Li, Xiu and Zhang, Weinan and Zhang, Chi and Bai, Chenjia and Li, Xuelong}, year={2026}, url={https://rhodes-team-prts.github.io} }




