遇见数据集

WarehousePick-v0_yolo

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

资源简介:

该数据集名为WarehousePick-v0_yolo,基于LeRobot框架创建,用于机器人仓库拾取任务。数据集包含40个片段(episodes),共计29707帧,任务数量为1。数据特征包括:6维动作(肩关节、肘关节、腕关节、夹爪等位置)、6维观察状态、三个摄像头视角的图像(innomaker、intel_rgb、front,分辨率分别为720x1280、424x240、720x1280,均为30fps的AV1视频),以及YOLO目标检测结果(最多20个检测框,每个框6维信息)和检测数量。机器人类型为so101_follower,数据划分为训练集(0-40个片段)。

The dataset is named WarehousePick-v0_yolo, created based on the LeRobot framework for robot warehouse picking tasks. It contains 40 episodes with a total of 29,707 frames and 1 task. Data features include: 6-dimensional actions (shoulder, elbow, wrist, gripper positions), 6-dimensional observation states, images from three camera views (innomaker, intel_rgb, front with resolutions 720x1280, 424x240, 720x1280 respectively, all 30fps AV1 videos), and YOLO object detection results (up to 20 bounding boxes, each with 6-dimensional information) and detection counts. The robot type is so101_follower, and the data is split into training set (episodes 0-40).

提供机构:
MrDuarte
创建时间:
2026-09-02
原始信息汇总

数据集概述:WarehousePick-v0_yolo

该数据集是一个用于机器人仓库取物任务(WarehousePick)的机器人学习数据集,使用LeRobot工具创建。

基本信息

  • 许可证: Apache 2.0
  • 任务类别: 机器人技术(robotics)
  • 标签: LeRobot

数据集内容与规模

  • 总片段数(Episodes): 40个
  • 总帧数(Frames): 29,707帧
  • 任务数: 1个
  • 帧率: 30 fps
  • 数据文件大小: 100 MB(Parquet格式)
  • 视频文件大小: 200 MB(MP4格式)
  • 机器人类型: so101_follower
  • 数据划分: 训练集包含全部40个片段

数据特征

数据集包含以下特征类型:

动作数据(action)

  • 6维浮点数组,对应机械臂关节和夹爪的位置指令
  • 包含:肩部旋转、肩部抬升、肘部弯曲、腕部弯曲、腕部旋转、夹爪

状态观测(observation.state)

  • 与动作对应的6维浮点数组,表示关节位置状态

多视角图像观测

数据集包含三个摄像头视角的视频数据:

视角名称 分辨率(高×宽) 视频编码 通道数
innomaker 720×1280 AV1 (yuv420p) 3
intel_rgb 424×240 AV1 (yuv420p) 3
front 720×1280 AV1 (yuv420p) 3

所有视频均以30 fps录制,无音频。

YOLO检测信息

  • yolo_detections: 20×6的浮点数组,存储检测结果
  • yolo_num_detections: 检测数量信息

时间与索引信息

  • timestamp: 时间戳(float32)
  • frame_index: 帧索引(int64)
  • episode_index: 片段索引(int64)
  • index: 索引(int64)
  • task_index: 任务索引(int64)

数据集结构

元数据文件(meta/info.json)记录数据集的详细信息,包括特征定义、代码库版本(v3.0)、数据路径结构等。数据以Parquet格式分块存储,视频以MP4格式按视角和分块组织。

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