wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add _image datasets in tests/data

This commit is contained in:
Remi Cadene 2024-05-17 14:50:13 +00:00
parent 7c5bd77d9c
commit c6fe57ede6
54 changed files with 654 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/data/lerobot/aloha_sim_insertion_human_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:e7dbc214a415689ca7fb83b6f8e12ec7824dfe34a66024b0b24bfeb3aeefd0e4
size 928

4
tests/data/lerobot/aloha_sim_insertion_human_image/meta_data/info.json Normal file
View File

@ -0,0 +1,4 @@
{
"fps": 50,
"video": 0
}

3
tests/data/lerobot/aloha_sim_insertion_human_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:bf78b400bae8dce9f966ca8d84edb4e00af003d3b2b19aac80c82c6375533720
size 2904

BIN
tests/data/lerobot/aloha_sim_insertion_human_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

47
tests/data/lerobot/aloha_sim_insertion_human_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,47 @@
{
"citation": "",
"description": "",
"features": {
"observation.images.top": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 14,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 14,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/aloha_sim_insertion_human_image/train/state.json Normal file
View File

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

3
tests/data/lerobot/aloha_sim_insertion_scripted_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:4500f31e62f0928a837fa71783acacda0db516c7b00d0586a41ea5fd8fc5e772
size 928

4
tests/data/lerobot/aloha_sim_insertion_scripted_image/meta_data/info.json Normal file
View File

@ -0,0 +1,4 @@
{
"fps": 50,
"video": 0
}

3
tests/data/lerobot/aloha_sim_insertion_scripted_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:9fba1eaa466afaa1b5fb86f69ab673cc194b916dfaa47b7051b89f7d73aeeffc
size 2904

BIN
tests/data/lerobot/aloha_sim_insertion_scripted_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

47
tests/data/lerobot/aloha_sim_insertion_scripted_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,47 @@
{
"citation": "",
"description": "",
"features": {
"observation.images.top": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 14,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 14,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/aloha_sim_insertion_scripted_image/train/state.json Normal file
View File

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

3
tests/data/lerobot/aloha_sim_transfer_cube_human_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:4500f31e62f0928a837fa71783acacda0db516c7b00d0586a41ea5fd8fc5e772
size 928

4
tests/data/lerobot/aloha_sim_transfer_cube_human_image/meta_data/info.json Normal file
View File

@ -0,0 +1,4 @@
{
"fps": 50,
"video": 0
}

3
tests/data/lerobot/aloha_sim_transfer_cube_human_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:a97f037f30c5c218e4b26e58bc556c5cd9c78b815c220613fd47cd5af7340037
size 2904

BIN
tests/data/lerobot/aloha_sim_transfer_cube_human_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

47
tests/data/lerobot/aloha_sim_transfer_cube_human_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,47 @@
{
"citation": "",
"description": "",
"features": {
"observation.images.top": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 14,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 14,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/aloha_sim_transfer_cube_human_image/train/state.json Normal file
View File

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

3
tests/data/lerobot/aloha_sim_transfer_cube_scripted_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:4500f31e62f0928a837fa71783acacda0db516c7b00d0586a41ea5fd8fc5e772
size 928

4
tests/data/lerobot/aloha_sim_transfer_cube_scripted_image/meta_data/info.json Normal file
View File

@ -0,0 +1,4 @@
{
"fps": 50,
"video": 0
}

3
tests/data/lerobot/aloha_sim_transfer_cube_scripted_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:2264d79241fbe28ff02c277ec7e15468513f79a5b03c34c264dcdd6c062c6cd2
size 2904

BIN
tests/data/lerobot/aloha_sim_transfer_cube_scripted_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

47
tests/data/lerobot/aloha_sim_transfer_cube_scripted_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,47 @@
{
"citation": "",
"description": "",
"features": {
"observation.images.top": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 14,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 14,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/aloha_sim_transfer_cube_scripted_image/train/state.json Normal file
View File

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

3
tests/data/lerobot/pusht_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:6ab094bd9de1a90273cefa1e02a84b3c3916e7812b9c73932f8f92ab2f1b9ba9
size 3432

4
tests/data/lerobot/pusht_image/meta_data/info.json Normal file
View File

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

3
tests/data/lerobot/pusht_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:65ceff2650ebbba2ee024be9ec083b7f8f20a69cd2c5bd6624382fe5fb974697
size 3056

BIN
tests/data/lerobot/pusht_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

55
tests/data/lerobot/pusht_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,55 @@
{
"citation": "",
"description": "",
"features": {
"observation.image": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 2,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 2,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.reward": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"next.success": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/pusht_image/train/state.json Normal file
View File

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

3
tests/data/lerobot/xarm_lift_medium_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:1ddb52e362094cc1469f34f7d723e235abccd24713b962f9765b4f910e85cebd
size 12936

4
tests/data/lerobot/xarm_lift_medium_image/meta_data/info.json Normal file
View File

@ -0,0 +1,4 @@
{
"fps": 15,
"video": 0
}

3
tests/data/lerobot/xarm_lift_medium_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:625213579edb7c82580ded28b3bb3dd6894bf27502b401a98de2908a65a2a15d
size 2808

BIN
tests/data/lerobot/xarm_lift_medium_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

51
tests/data/lerobot/xarm_lift_medium_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,51 @@
{
"citation": "",
"description": "",
"features": {
"observation.image": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 4,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 3,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.reward": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/xarm_lift_medium_image/train/state.json Normal file
View File

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

3
tests/data/lerobot/xarm_lift_medium_replay_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:1ddb52e362094cc1469f34f7d723e235abccd24713b962f9765b4f910e85cebd
size 12936

4
tests/data/lerobot/xarm_lift_medium_replay_image/meta_data/info.json Normal file
View File

@ -0,0 +1,4 @@
{
"fps": 15,
"video": 0
}

3
tests/data/lerobot/xarm_lift_medium_replay_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:625213579edb7c82580ded28b3bb3dd6894bf27502b401a98de2908a65a2a15d
size 2808

BIN
tests/data/lerobot/xarm_lift_medium_replay_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

51
tests/data/lerobot/xarm_lift_medium_replay_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,51 @@
{
"citation": "",
"description": "",
"features": {
"observation.image": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 4,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 3,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.reward": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/xarm_lift_medium_replay_image/train/state.json Normal file
View File

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

3
tests/data/lerobot/xarm_push_medium_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:1ddb52e362094cc1469f34f7d723e235abccd24713b962f9765b4f910e85cebd
size 12936

4
tests/data/lerobot/xarm_push_medium_image/meta_data/info.json Normal file
View File

@ -0,0 +1,4 @@
{
"fps": 15,
"video": 0
}

3
tests/data/lerobot/xarm_push_medium_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:625213579edb7c82580ded28b3bb3dd6894bf27502b401a98de2908a65a2a15d
size 2808

BIN
tests/data/lerobot/xarm_push_medium_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

51
tests/data/lerobot/xarm_push_medium_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,51 @@
{
"citation": "",
"description": "",
"features": {
"observation.image": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 4,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 3,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.reward": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/xarm_push_medium_image/train/state.json Normal file
View File

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

3
tests/data/lerobot/xarm_push_medium_replay_image/meta_data/episode_data_index.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:1ddb52e362094cc1469f34f7d723e235abccd24713b962f9765b4f910e85cebd
size 12936

4
tests/data/lerobot/xarm_push_medium_replay_image/meta_data/info.json Normal file
View File

@ -0,0 +1,4 @@
{
"fps": 15,
"video": 0
}

3
tests/data/lerobot/xarm_push_medium_replay_image/meta_data/stats.safetensors Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:625213579edb7c82580ded28b3bb3dd6894bf27502b401a98de2908a65a2a15d
size 2808

BIN
tests/data/lerobot/xarm_push_medium_replay_image/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

51
tests/data/lerobot/xarm_push_medium_replay_image/train/dataset_info.json Normal file
View File

@ -0,0 +1,51 @@
{
"citation": "",
"description": "",
"features": {
"observation.image": {
"_type": "Image"
},
"observation.state": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 4,
"_type": "Sequence"
},
"action": {
"feature": {
"dtype": "float32",
"_type": "Value"
},
"length": 3,
"_type": "Sequence"
},
"episode_index": {
"dtype": "int64",
"_type": "Value"
},
"frame_index": {
"dtype": "int64",
"_type": "Value"
},
"timestamp": {
"dtype": "float32",
"_type": "Value"
},
"next.reward": {
"dtype": "float32",
"_type": "Value"
},
"next.done": {
"dtype": "bool",
"_type": "Value"
},
"index": {
"dtype": "int64",
"_type": "Value"
}
},
"homepage": "",
"license": ""
}

13
tests/data/lerobot/xarm_push_medium_replay_image/train/state.json Normal file
View File

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