遇见数据集

unity_niryo_pi0_dataset

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

资源简介:

这是一个用于机器人学习的数据集,由LeRobot工具创建,基于Niryo机器人类型。数据集包含4个episodes,总计1010帧,帧率为30fps,用于单个任务。数据以parquet文件格式存储,总数据文件大小为100MB,视频文件大小为200MB。特征包括图像(尺寸3x224x224)、手腕图像(尺寸3x224x224)、状态(6维浮点数组)、动作(6维浮点数组)、时间戳、帧索引、episode索引、索引和任务索引。数据集仅包含训练分割(episodes 0到4),适用于机器人控制和视觉任务的研究。

This is a dataset for robot learning, created by the LeRobot tool and based on the Niryo robot type. The dataset contains 4 episodes, totaling 1010 frames, with a frame rate of 30fps, for a single task. Data is stored in parquet file format, with a total data file size of 100MB and a video file size of 200MB. Features include images (size 3x224x224), wrist images (size 3x224x224), state (6-dimensional float array), action (6-dimensional float array), timestamp, frame index, episode index, index, and task index. The dataset only includes a training split (episodes 0 to 4) and is suitable for research in robot control and vision tasks.

提供机构:
Gian26
创建时间:
2026-05-29
原始信息汇总

数据集概述

  • 名称: unity_niryo_pi0_dataset
  • 许可协议: Apache-2.0
  • 任务类型: 机器人学 (Robotics)
  • 标签: LeRobot

数据集描述

该数据集使用 LeRobot 创建,专注于机器人操控数据。数据集包含 4 个片段(episodes),总计 1010 帧,覆盖 1 个任务。训练集包含所有片段(索引 0 到 3)。

数据集结构

  • 数据格式: 以 Parquet 文件存储(位于 data/*/*.parquet),另有视频文件(MP4 格式)存储在 videos/ 目录。
  • 机器人类型: Niryo
  • 帧率 (FPS): 30
  • 数据文件大小: 约 100 MB
  • 视频文件大小: 约 200 MB
  • 数据分块: 每块包含 1000 帧(chunks_size)

特征字段

字段名 数据类型 形状 描述
image 图像 (image) (3, 224, 224) 主摄像头图像
wrist_image 图像 (image) (3, 224, 224) 腕部摄像头图像
state 浮点数 (float32) (6,) 机器人状态(6维)
actions 浮点数 (float32) (6,) 动作指令(6维)
timestamp 浮点数 (float32) (1,) 时间戳
frame_index 整数 (int64) (1,) 帧索引
episode_index 整数 (int64) (1,) 片段索引
index 整数 (int64) (1,) 索引
task_index 整数 (int64) (1,) 任务索引

数据集可视化

可通过 此链接 在线可视化数据集内容。

引用信息

目前缺少 BibTeX 引用信息。

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