Datasetpublic

FedorX8/berkeley_rpt_lerobot

This dataset was created using LeRobot. Dataset Structure meta/info.json: { "codebase_version": "v3.0", "robot_type": "unknown", "total_episodes": 908, "total_frames": 392578, "total_tasks": 4, "chunks_size": 1000, "fps": 30, "splits": { "train": "0:908" }, "data_path": "data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet", "video_path": "videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4", "features": {… See the full description on the dataset page: https://huggingface.co/datasets/FedorX8/berkeley_rpt_lerobot.

sourceHugging Facemitupdated 7mo agoView on Hugging Face

0likes33downloads

Dataset Card

This dataset was created using LeRobot.

Dataset Description

—Homepage: [More Information Needed]
—Paper: https://arxiv.org/abs/2306.10007
—License: mit

Dataset Structure

meta/info.json:

json

{
  "codebase_version": "v3.0",
  "robot_type": "unknown",
  "total_episodes": 908,
  "total_frames": 392578,
  "total_tasks": 4,
  "chunks_size": 1000,
  "fps": 30,
  "splits": {
    "train": "0:908"
  },
  "data_path": "data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet",
  "video_path": "videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4",
  "features": {
    "observation.images.hand_image": {
      "dtype": "video",
      "shape": [
        480,
        640,
        3
      ],
      "names": [
        "height",
        "width",
        "channel"
      ],
      "video_info": {
        "video.fps": 30.0,
        "video.codec": "av1",
        "video.pix_fmt": "yuv420p",
        "video.is_depth_map": false,
        "has_audio": false
      }
    },
    "observation.state": {
      "dtype": "float32",
      "shape": [
        8
      ],
      "names": {
        "motors": [
          "motor_0",
          "motor_1",
          "motor_2",
          "motor_3",
          "motor_4",
          "motor_5",
          "motor_6",
          "motor_7"
        ]
      },
      "fps": 30.0
    },
    "action": {
      "dtype": "float32",
      "shape": [
        8
      ],
      "names": {
        "motors": [
          "motor_0",
          "motor_1",
          "motor_2",
          "motor_3",
          "motor_4",
          "motor_5",
          "motor_6",
          "motor_7"
        ]
      },
      "fps": 30.0
    },
    "timestamp": {
      "dtype": "float32",
      "shape": [
        1
      ],
      "names": null,
      "fps": 30.0
    },
    "episode_index": {
      "dtype": "int64",
      "shape": [
        1
      ],
      "names": null,
      "fps": 30.0
    },
    "frame_index": {
      "dtype": "int64",
      "shape": [
        1
      ],
      "names": null,
      "fps": 30.0
    },
    "next.reward": {
      "dtype": "float32",
      "shape": [
        1
      ],
      "names": null,
      "fps": 30.0
    },
    "next.done": {
      "dtype": "bool",
      "shape": [
        1
      ],
      "names": null,
      "fps": 30.0
    },
    "index": {
      "dtype": "int64",
      "shape": [
        1
      ],
      "names": null,
      "fps": 30.0
    },
    "task_index": {
      "dtype": "int64",
      "shape": [
        1
      ],
      "names": null,
      "fps": 30.0
    }
  },
  "data_files_size_in_mb": 100,
  "video_files_size_in_mb": 500
}

Citation

BibTeX:

bibtex

@article{Radosavovic2023,
    title={Robot Learning with Sensorimotor Pre-training},
    author={Ilija Radosavovic and Baifeng Shi and Letian Fu and Ken Goldberg and Trevor Darrell and Jitendra Malik},
    year={2023},
    journal={arXiv:2306.10007}
}