pusht_video

该数据集由LeRobot创建，属于机器人领域。数据集包含206个episodes，25650帧，1个任务。数据文件格式为parquet，视频文件格式为mp4。特征包括图像观察（96x96x3）、状态观察（2个电机状态）、动作（2个电机控制）、奖励、完成标志、成功标志等。数据集的帧率为10fps，总数据文件大小为100MB，视频文件大小为500MB。

原始信息汇总

数据集概述

基本信息

• 数据集名称: szk1ck/pusht_video
• 创建工具: LeRobot
• 许可证: Apache-2.0
• 任务类别: 机器人学

数据集规模

• 总情节数: 206
• 总帧数: 25650
• 总任务数: 1
• 数据文件大小: 100 MB
• 视频文件大小: 500 MB
• 帧率: 10 FPS
• 数据分块大小: 1000

数据划分

• 训练集: 包含全部206个情节。

数据结构

数据以Parquet文件格式存储，路径模式为 data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet。 视频以MP4文件格式存储，路径模式为 videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4。

数据特征

• observation.image
  • 类型: 视频
  • 形状: [96, 96, 3] (高度，宽度，通道)
  • 帧率: 10.0
  • 视频信息: 高度96像素，宽度96像素，编码格式为av1，像素格式为yuv420p，非深度图，3个通道，无音频。
• observation.state
  • 类型: float32
  • 形状: [2]
  • 名称: ["motor_0", "motor_1"]
  • 帧率: 10.0
• action
  • 类型: float32
  • 形状: [2]
  • 名称: ["motor_0", "motor_1"]
  • 帧率: 10.0
• episode_index
  • 类型: int64
  • 形状: [1]
• frame_index
  • 类型: int64
  • 形状: [1]
• timestamp
  • 类型: float32
  • 形状: [1]
• next.reward
  • 类型: float32
  • 形状: [1]
  • 帧率: 10.0
• next.done
  • 类型: bool
  • 形状: [1]
  • 帧率: 10.0
• next.success
  • 类型: bool
  • 形状: [1]
  • 帧率: 10.0
• index
  • 类型: int64
  • 形状: [1]
• task_index
  • 类型: int64
  • 形状: [1]

可视化

可通过以下链接可视化数据集：https://huggingface.co/spaces/lerobot/visualize_dataset?path=szk1ck/pusht_video

引用信息

• 主页: 信息缺失
• 论文: 信息缺失
• BibTeX引用: 信息缺失