遇见数据集

hang_mug_on_tree

收藏
Hugging Face2026-06-06 更新2026-06-06 收录
官方服务:

资源简介:

该数据集是一个用于机器人学任务的双臂机器人数据集,使用LeRobot工具创建。数据集包含200个episodes,总计98174帧数据,涉及1个任务。数据以30帧每秒的速率采集,存储为parquet格式文件,并包含600个视频文件。数据集的特征包括观察状态(32维浮点数组,表示左右机械臂的位置、旋转6D表示、夹爪状态和关节角度)、动作(与观察状态相同的32维浮点数组)、来自三个视角的图像观察(头部、左手腕和右手腕,均为3通道360x640分辨率视频),以及时间戳、帧索引、episode索引等元数据。数据集适用于机器人控制、模仿学习等研究。

This dataset is a dual-arm robot dataset for robotics tasks, created using the LeRobot tool. It contains 200 episodes, totaling 98174 frames of data, covering 1 task. The data is collected at a rate of 30 frames per second, stored in parquet format files, and includes 600 video files. The dataset features include observation states (a 32-dimensional floating-point array representing the positions, rotation 6D representations, gripper states, and joint angles of the left and right arms), actions (the same 32-dimensional array as the observation states), image observations from three perspectives (head, left wrist, and right wrist, all as 3-channel 360x640 resolution videos), and metadata such as timestamps, frame indices, and episode indices. The dataset is suitable for research in robot control, imitation learning, and related areas.

提供机构:
YOLO2431
创建时间:
2026-06-06
原始信息汇总

数据集概述

  • 数据集名称: YOLO2431/hang_mug_on_tree
  • 许可证: Apache-2.0
  • 任务类别: 机器人学(Robotics)
  • 标签: LeRobot

数据集详情

创建工具

该数据集使用 LeRobot 创建。

数据规模

  • 总片段数(Episodes): 200
  • 总帧数(Frames): 98,174
  • 总任务数: 1
  • 总视频数: 600
  • 总块数(Chunks): 1
  • 块大小: 1,000
  • 帧率(FPS): 30
  • 数据划分: 仅训练集(train: 0-200),无验证集或测试集

机器人类型

  • 机器人型号: yam_bimanual(双臂机器人)

数据结构

数据文件与视频路径

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

特征(Features)

数据集包含以下特征字段:

特征名称 数据类型 形状 说明
observation.state float32 (32,) 机器人观察状态,包含左右臂位置、旋转、夹爪及关节数据
action float32 (32,) 机器人动作,与状态维度相同
observation.images.head video (3, 360, 640) 头部摄像头视频,H.264编码,30 FPS
observation.images.left_wrist video (3, 360, 640) 左腕摄像头视频,H.264编码,30 FPS
observation.images.right_wrist video (3, 360, 640) 右腕摄像头视频,H.264编码,30 FPS
timestamp float32 (1,) 时间戳
frame_index int64 (1,) 帧索引
episode_index int64 (1,) 片段索引
index int64 (1,) 全局索引
task_index int64 (1,) 任务索引

状态与动作的32维详细构成

  • 左臂位置(3维):自由度位置 x, y, z
  • 左臂旋转(6维):6D旋转表示
  • 右臂位置(3维):自由度位置 x, y, z
  • 右臂旋转(6维):6D旋转表示
  • 左夹爪(1维)
  • 右夹爪(1维)
  • 左关节(6维):关节角度
  • 右关节(6维):关节角度

视频属性

  • 分辨率: 640x360
  • 编码格式: H.264 (yuv420p)
  • 帧率: 30 FPS
  • 通道数: 3 (RGB)
  • 无音频

引用信息

当前无可用引用文献。

二维码
社区交流群
二维码
科研交流群
商业服务