five

grasp_red_cube_2c

收藏
Hugging Face2025-08-23 更新2025-08-24 收录
下载链接:
https://huggingface.co/datasets/K1t1k/grasp_red_cube_2c
下载链接
链接失效反馈
官方服务:
资源简介:
这是一个由LeRobot创建的数据集,包含10个episodes,4926帧图像,针对1个任务,共有20个视频文件。数据集提供了动作、状态、图像等多种特征,适用于机器人学相关任务。数据以Parquet格式存储,视频文件采用av1编码。
创建时间:
2025-08-16
原始信息汇总

数据集概述

基本信息

  • 名称:grasp_red_cube_2c
  • 许可证:Apache-2.0
  • 任务类别:机器人技术
  • 标签:LeRobot

数据集结构

  • 总任务数:1
  • 总视频数:20
  • 总片段数:10
  • 总帧数:4926
  • 块大小:1000
  • 帧率:30 fps
  • 分割:训练集(0:10)

数据格式

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

特征描述

  • 动作(action):浮点32,形状[6],包含肩部平移、肩部提升、肘部弯曲、腕部弯曲、腕部旋转和夹爪位置
  • 观测状态(observation.state):浮点32,形状[6],关节位置信息
  • 观测图像(observation.images.up):视频格式,分辨率480×640×3,AV1编码,无音频
  • 观测图像(observation.images.front):视频格式,分辨率480×640×3,AV1编码,无音频
  • 时间戳(timestamp):浮点32,形状[1]
  • 帧索引(frame_index):整型64,形状[1]
  • 片段索引(episode_index):整型64,形状[1]
  • 索引(index):整型64,形状[1]
  • 任务索引(task_index):整型64,形状[1]

机器人信息

  • 机器人类型:so101_follower
  • 代码库版本:v2.1

引用信息

  • 主页:[更多信息需要补充]
  • 论文:[更多信息需要补充]
  • BibTeX引用:[更多信息需要补充]
搜集汇总
数据集介绍
main_image_url
构建方式
在机器人操作任务研究领域,grasp_red_cube_2c数据集通过LeRobot平台系统构建,采用SO101型机械臂执行抓取红色立方体的标准化操作。数据采集过程包含10个完整操作片段,以30fps帧率同步记录多模态数据,包括六维关节角度状态、双视角视觉影像及时间序列标记,最终以标准化Parquet格式分块存储,确保数据结构的完整性与可追溯性。
特点
该数据集的核心特征体现在其多模态同步采集架构,同时涵盖机械臂的六自由度动作指令、实时关节状态反馈及双视角(上方与前方)RGB视频流。所有数据均附带精确的时间戳与帧索引,支持端到端的动作-状态-视觉关联分析。数据集规模包含4926帧有效数据,视频分辨率达640x480,采用AV1编码压缩,兼具高精度与存储效率,为模仿学习与行为克隆研究提供丰富样本。
使用方法
研究者可通过加载Parquet数据文件直接访问结构化数据流,其中动作与状态数据以float32格式存储,视频数据则通过路径索引调用。数据集默认按训练集划分,支持按帧索引或片段索引提取子序列。使用时应注重时序一致性校验,建议结合LeRobot平台工具进行数据可视化与预处理,以充分发挥多模态数据在机器人策略学习中的协同效应。
背景与挑战
背景概述
机器人操作任务数据集grasp_red_cube_2c由LeRobot研究团队基于Apache 2.0开源协议构建,专注于机械臂抓取操作的示范学习领域。该数据集采用so101_follower型机器人平台,通过多视角视觉感知与六维关节控制数据,记录了完整的抓取红色立方体任务序列。其构建体现了示范学习在机器人精细操作中的关键价值,为模仿学习与强化学习算法提供了高质量的多模态训练数据,推动了机器人自主操作能力的发展。
当前挑战
该数据集核心挑战在于解决高维连续动作空间中的精确抓取问题,需协调多关节运动与视觉感知的时序对齐。构建过程中面临多传感器数据同步的技术难题,包括双视角视频流与机械臂状态数据的时间戳精确匹配。此外,示范数据采集需保证操作轨迹的平滑性与任务成功率,对机器人控制精度与数据采集稳定性提出极高要求。
常用场景
经典使用场景
在机器人操作学习领域,该数据集通过记录六轴机械臂抓取红色立方体的完整运动轨迹,为模仿学习算法提供高质量示范数据。其多视角视觉观测与关节位置动作的同步记录,使得研究者能够训练端到端的视觉运动策略网络,实现从像素输入到关节控制的直接映射。
解决学术问题
该数据集有效解决了机器人操作任务中示范数据稀缺性与真实性问题,为行为克隆、逆强化学习等算法提供标准化评测基准。通过精确记录机械臂关节角度与双视角视觉信息,它助力研究者突破视觉-运动协同建模的技术瓶颈,推动机器人精细操作能力的理论发展。
衍生相关工作
该数据集催生了多项基于Transformer的视觉运动建模研究,其中时空注意力机制被应用于长序列动作预测。部分工作通过提取双视角视觉特征融合表示,提升了抓取位姿估计的精度;另有研究利用其多模态特性开发了跨视角视觉一致性约束的自监督学习方法。
以上内容由遇见数据集搜集并总结生成
5,000+
优质数据集
54 个
任务类型
进入经典数据集
二维码
社区交流群

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

二维码
科研交流群

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

数据驱动未来

携手共赢发展

商业合作