five

coop_multitask_dataset

收藏
Hugging Face2025-10-25 更新2025-10-26 收录
下载链接:
https://huggingface.co/datasets/lucanunz/coop_multitask_dataset
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集是一个使用LeRobot工具创建的机器人数据集,包含了一个名为panda的机器人类型。数据集共有4个剧集,391个帧,4个任务,没有视频文件,只有一个数据块。数据集的帧率为10fps,并且提供了训练集的拆分信息。数据集包含了多种特征,如图像、手腕图像、状态、动作、环境状态、变换矩阵、活跃机器人、指令动作、TCP状态、夹爪状态、夹爪指令、位置、时间戳、帧索引、剧集索引和任务索引。
创建时间:
2025-10-25
原始信息汇总

coop_multitask_dataset 数据集概述

基本信息

  • 许可证: Apache-2.0
  • 任务类别: 机器人学
  • 标签: LeRobot, panda, coop_env, multitask
  • 创建工具: LeRobot

数据集规模

  • 总任务数: 4
  • 总片段数: 4
  • 总帧数: 391
  • 总视频数: 0
  • 总数据块数: 1
  • 数据块大小: 1000
  • 帧率: 10 fps

数据结构

数据划分

  • 训练集: 0:4

数据格式

  • 数据文件: Parquet格式
  • 数据路径: data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet
  • 视频路径: videos/chunk-{episode_chunk:03d}/{video_key}/episode_{episode_index:06d}.mp4

特征字段

图像特征

  • image: 256×256×3图像
  • wrist_image: 256×256×3腕部图像

状态特征

  • state: 7维浮点状态向量
  • env_state: 156维浮点环境状态向量
  • qpos: 7维浮点关节位置向量
  • tcp_states: 12维浮点TCP状态向量

动作特征

  • actions: 7维浮点动作向量
  • commanded_actions: 12维浮点指令动作向量
  • gripper_commands: 1维浮点夹爪指令

变换矩阵

  • transformation_matrices: 16维浮点变换矩阵

机器人状态

  • active_robot: 1维整型激活机器人标识
  • gripper_states: 1维浮点夹爪状态

索引特征

  • timestamp: 1维浮点时间戳
  • frame_index: 1维整型帧索引
  • episode_index: 1维整型片段索引
  • index: 1维整型索引
  • task_index: 1维整型任务索引

技术规格

  • 代码库版本: v2.1
  • 机器人类型: panda
  • 数据类型: float32, int64, image

引用信息

  • 主页: 待补充
  • 论文: 待补充
  • BibTeX引用: 待补充
搜集汇总
数据集介绍
main_image_url
构建方式
在机器人协作环境研究领域,coop_multitask_dataset的构建依托LeRobot开源框架实现。该数据集通过Panda机器人平台采集多任务交互数据,采用分块存储机制将391帧数据划分为4个完整 episodes,并以10fps的采样频率记录传感器信息。数据以标准化Parquet格式保存,每个数据块包含机器人状态、动作指令及环境观测等多模态特征,确保了数据结构的完整性与可扩展性。
特点
该数据集的核心价值体现在其多维度的机器人交互特征体系。除了包含双视角256×256像素的图像流(image与wrist_image),还完整记录了7自由度机械臂状态、12维末端执行器轨迹、156维环境状态及16维变换矩阵等物理量。特别值得关注的是其融合了主动机器人标识、抓取器控制指令等协作任务特有参数,为研究多智能体协同操作提供了细粒度的数据支撑。
使用方法
针对机器人模仿学习与多任务策略研究,该数据集支持通过标准数据加载接口直接调用。研究者可依据episode索引访问不同任务序列,利用帧索引实现时序数据分析。数据集已预置训练集划分,可直接用于行为克隆、逆强化学习等算法验证。各传感器数据通道均具备明确的维度标注,便于构建端到端的感知控制模型。
背景与挑战
背景概述
在机器人学习领域,多任务协同操作已成为提升智能体泛化能力的关键路径。Coop Multitask Dataset基于LeRobot开源框架构建,采用Franka Emika Panda机械臂作为实验平台,聚焦于协作环境下的多任务学习范式。该数据集通过集成视觉感知模块与关节状态控制信号,记录了包含391帧序列的4个独立任务场景,其多维特征空间涵盖图像观测、机械臂状态及环境交互矩阵等16类结构化数据,为研究跨任务知识迁移提供了标准化基准。
当前挑战
该数据集致力于解决机器人多任务协同操作中的动作泛化难题,需克服异质任务间策略共享的语义鸿沟。构建过程中面临传感器时序对齐的技术瓶颈,包括7自由度机械臂轨迹与双视角视觉数据(256×256×3分辨率)的精确同步。此外,156维环境状态与12维指令动作的耦合关系建模,对数据采集系统的实时性与完整性提出了严苛要求,而多模态特征的空间尺度差异进一步加剧了表征学习的复杂性。
常用场景
经典使用场景
在机器人学习领域,coop_multitask_dataset作为多任务协作环境下的标准基准,主要用于训练和评估机械臂在复杂场景中的决策能力。该数据集通过集成视觉感知与关节控制数据,支持从原始图像输入到动作输出的端到端策略学习,典型应用于模仿学习与强化学习算法的性能验证。
解决学术问题
该数据集有效解决了机器人泛化能力不足的核心难题,通过多任务序列数据揭示了状态-动作映射的潜在规律。其丰富的传感器模态(如双视角图像与七维关节状态)为部分可观测马尔可夫决策过程的理论研究提供了实验基础,显著推进了跨任务知识迁移与长期动作规划等前沿方向的探索。
衍生相关工作
以LeRobot开源框架为基石,该数据集催生了系列创新研究。典型衍生工作包括基于时空特征融合的多任务策略蒸馏方法、结合变换矩阵的视觉运动对齐模型,以及利用任务索引实现分层强化学习的架构设计,这些成果持续推动着协作机器人认知能力的发展边界。
以上内容由遇见数据集搜集并总结生成
5,000+
优质数据集
54 个
任务类型
进入经典数据集
二维码
社区交流群

面向社区/商业的数据集话题

二维码
科研交流群

面向高校/科研机构的开源数据集话题

数据驱动未来

携手共赢发展

商业合作