FedorX8/bridge_v2_lerobot

---
license: apache-2.0
task_categories:
- robotics
tags:
- LeRobot
configs:
- config_name: default
data_files: data/*/*.parquet
---

This dataset was created using [LeRobot](https://github.com/huggingface/lerobot).

## Dataset Description

Converted BridgeV2 dataset to lerobot format.
Downloaded from [https://rail.eecs.berkeley.edu/datasets/bridge_release/data/] (only demos_8_17.zip, not scripted_6_18.zip)

## Dataset Structure

[meta/info.json](meta/info.json):
```json
{
"codebase_version": "v3.0",
"robot_type": "widowx",
"total_episodes": 50418,
"total_frames": 1801162,
"total_tasks": 22199,
"chunks_size": 1000,
"data_files_size_in_mb": 100,
"video_files_size_in_mb": 200,
"fps": 5,
"splits": {
"train": "0:50418"
},
"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.primary": {
"dtype": "video",
"shape": [
480,
640,
3
],
"names": [
"height",
"width",
"channels"
],
"info": {
"video.height": 480,
"video.width": 640,
"video.codec": "av1",
"video.pix_fmt": "yuv420p",
"video.is_depth_map": false,
"video.fps": 5,
"video.channels": 3,
"has_audio": false
}
},
"observation.images.wrist": {
"dtype": "video",
"shape": [
480,
640,
3
],
"names": [
"height",
"width",
"channels"
],
"info": {
"video.height": 480,
"video.width": 640,
"video.codec": "av1",
"video.pix_fmt": "yuv420p",
"video.is_depth_map": false,
"video.fps": 5,
"video.channels": 3,
"has_audio": false
}
},
"observation.images.side1": {
"dtype": "video",
"shape": [
480,
640,
3
],
"names": [
"height",
"width",
"channels"
],
"info": {
"video.height": 480,
"video.width": 640,
"video.codec": "av1",
"video.pix_fmt": "yuv420p",
"video.is_depth_map": false,
"video.fps": 5,
"video.channels": 3,
"has_audio": false
}
},
"observation.images.side2": {
"dtype": "video",
"shape": [
480,
640,
3
],
"names": [
"height",
"width",
"channels"
],
"info": {
"video.height": 480,
"video.width": 640,
"video.codec": "av1",
"video.pix_fmt": "yuv420p",
"video.is_depth_map": false,
"video.fps": 5,
"video.channels": 3,
"has_audio": false
}
},
"observation.state": {
"dtype": "float32",
"shape": [
7
],
"names": {
"axes": [
"joint_0",
"joint_1",
"joint_2",
"joint_3",
"joint_4",
"joint_5",
"gripper"
]
}
},
"action": {
"dtype": "float32",
"shape": [
7
],
"names": {
"axes": [
"action_0",
"action_1",
"action_2",
"action_3",
"action_4",
"action_5",
"gripper"
]
}
},
"language_instruction": {
"dtype": "string",
"shape": [
1
],
"names": null
},
"date": {
"dtype": "string",
"shape": [
1
],
"names": null
},
"timestamp": {
"dtype": "float32",
"shape": [
1
],
"names": null
},
"frame_index": {
"dtype": "int64",
"shape": [
1
],
"names": null
},
"episode_index": {
"dtype": "int64",
"shape": [
1
],
"names": null
},
"index": {
"dtype": "int64",
"shape": [
1
],
"names": null
},
"task_index": {
"dtype": "int64",
"shape": [
1
],
"names": null
}
}
}
```

## Citation

**BibTeX:**

```bibtex
@inproceedings{walke2023bridgedata,
title={BridgeData V2: A Dataset for Robot Learning at Scale},
author={Walke, Homer and Black, Kevin and Lee, Abraham and Kim, Moo Jin and Du, Max and Zheng, Chongyi and Zhao, Tony and Hansen-Estruch, Philippe and Vuong, Quan and He, Andre and Myers, Vivek and Fang, Kuan and Finn, Chelsea and Levine, Sergey},
booktitle={Conference on Robot Learning (CoRL)},
year={2023}
}
```

许可证：Apache-2.0
任务类别：
- 机器人学
标签：
- LeRobot
配置项：
- 配置名称：default
数据文件路径：data/*/*.parquet

本数据集由[LeRobot](https://github.com/huggingface/lerobot)开发构建。

## 数据集描述
本数据集将BridgeV2数据集转换为LeRobot格式，数据源自[https://rail.eecs.berkeley.edu/datasets/bridge_release/data/]，仅下载了demos_8_17.zip，未包含scripted_6_18.zip。

## 数据集结构
[meta/info.json](meta/info.json)：
json
{
"代码库版本": "v3.0",
"机器人类型": "widowx",
"总回合数": 50418,
"总帧数": 1801162,
"总任务数": 22199,
"分块大小": 1000,
"数据文件总大小（MB）": 100,
"视频文件总大小（MB）": 200,
"帧率": 5,
"数据集划分": {
"训练集": "0:50418"
},
"数据路径格式": "data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet",
"视频路径格式": "videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4",
"数据特征": {
"观测数据.图像.主摄像头": {
"数据类型": "视频",
"形状": [480, 640, 3],
"维度名称": ["高度", "宽度", "通道数"],
"详细信息": {
"视频高度": 480,
"视频宽度": 640,
"视频编码格式": "av1",
"视频像素格式": "yuv420p",
"是否为深度图": false,
"视频帧率": 5,
"视频通道数": 3,
"含音频": false
}
},
"观测数据.图像.腕部摄像头": {
"数据类型": "视频",
"形状": [480, 640, 3],
"维度名称": ["高度", "宽度", "通道数"],
"详细信息": {
"视频高度": 480,
"视频宽度": 640,
"视频编码格式": "av1",
"视频像素格式": "yuv420p",
"是否为深度图": false,
"视频帧率": 5,
"视频通道数": 3,
"含音频": false
}
},
"观测数据.图像.侧视摄像头1": {
"数据类型": "视频",
"形状": [480, 640, 3],
"维度名称": ["高度", "宽度", "通道数"],
"详细信息": {
"视频高度": 480,
"视频宽度": 640,
"视频编码格式": "av1",
"视频像素格式": "yuv420p",
"是否为深度图": false,
"视频帧率": 5,
"视频通道数": 3,
"含音频": false
}
},
"观测数据.图像.侧视摄像头2": {
"数据类型": "视频",
"形状": [480, 640, 3],
"维度名称": ["高度", "宽度", "通道数"],
"详细信息": {
"视频高度": 480,
"视频宽度": 640,
"视频编码格式": "av1",
"视频像素格式": "yuv420p",
"是否为深度图": false,
"视频帧率": 5,
"视频通道数": 3,
"含音频": false
}
},
"观测数据.状态": {
"数据类型": "float32",
"形状": [7],
"维度名称": {
"轴名称": ["关节0", "关节1", "关节2", "关节3", "关节4", "关节5", "夹爪"]
}
},
"动作指令": {
"数据类型": "float32",
"形状": [7],
"维度名称": {
"轴名称": ["动作0", "动作1", "动作2", "动作3", "动作4", "动作5", "夹爪"]
}
},
"语言指令": {
"数据类型": "string",
"形状": [1],
"维度名称": null
},
"日期": {
"数据类型": "string",
"形状": [1],
"维度名称": null
},
"时间戳": {
"数据类型": "float32",
"形状": [1],
"维度名称": null
},
"帧索引": {
"数据类型": "int64",
"形状": [1],
"维度名称": null
},
"回合索引": {
"数据类型": "int64",
"形状": [1],
"维度名称": null
},
"全局索引": {
"数据类型": "int64",
"形状": [1],
"维度名称": null
},
"任务索引": {
"数据类型": "int64",
"形状": [1],
"维度名称": null
}
}
}

## 引用
** BibTeX格式引用：**
bibtex
@inproceedings{walke2023bridgedata,
title={《BridgeData V2：面向大规模机器人学习的数据集》},
author={Walke, Homer and Black, Kevin and Lee, Abraham and Kim, Moo Jin and Du, Max and Zheng, Chongyi and Zhao, Tony and Hansen-Estruch, Philippe and Vuong, Quan and He, Andre and Myers, Vivek and Fang, Kuan and Finn, Chelsea and Levine, Sergey},
booktitle={《机器人学习会议（CoRL）》},
year={2023}
}