Add UMI-gripper dataset (#83)

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Adil Zouitine
2024-04-28 18:41:07 +02:00
committed by GitHub
parent a4b6c5e3b1
commit 81e490d46f
11 changed files with 706 additions and 29 deletions

View File

@@ -0,0 +1,3 @@
{
"fps": 10
}

View File

@@ -0,0 +1,67 @@
{
"citation": "",
"description": "",
"features": {
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 7,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"episode_data_index_from": {
"dtype": "int64",
"_type": "Value"
},
"episode_data_index_to": {
"dtype": "int64",
"_type": "Value"
},
"end_pose": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 6,
"_type": "Sequence"
},
"start_pos": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 6,
"_type": "Sequence"
},
"gripper_width": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 1,
"_type": "Sequence"
},
"index": {
"dtype": "int64",
"_type": "Value"
},
"observation.image": {
"_type": "Image"
}
},
"homepage": "",
"license": ""
}

View File

@@ -0,0 +1,13 @@
{
"_data_files": [
{
"filename": "data-00000-of-00001.arrow"
}
],
"_fingerprint": "fd95ee932cb1fce2",
"_format_columns": null,
"_format_kwargs": {},
"_format_type": "torch",
"_output_all_columns": false,
"_split": null
}