arnold_baseline / dataset_sample_report.json
Whalswp's picture
Upload folder using huggingface_hub
6ae525f verified
{
"camera_keys_present": {
"observation.images.base": true,
"observation.images.front": true,
"observation.images.wrist": true
},
"camera_order": [
"observation.images.front",
"observation.images.base",
"observation.images.wrist"
],
"keys": [
"action",
"action_is_pad",
"episode_index",
"frame_index",
"index",
"observation.ee_pose",
"observation.images.base",
"observation.images.front",
"observation.images.left",
"observation.images.wrist",
"observation.images.wrist_bottom",
"observation.state",
"task",
"task_index",
"timestamp"
],
"sample_index": 0,
"task": "close the cabinet entirely closed",
"task_index": {
"preview": [
0
],
"shape": []
},
"values": {
"action": {
"dtype": "torch.float32",
"preview": {
"preview": [
0.045190244913101196,
-0.6859986186027527,
0.07602229714393616,
-2.728437662124634,
0.015480773523449898
],
"shape": [
16,
9
]
},
"shape": [
16,
9
],
"type": "Tensor"
},
"action_is_pad": {
"dtype": "torch.bool",
"preview": {
"preview": [
false,
false,
false,
false,
false
],
"shape": [
16
]
},
"shape": [
16
],
"type": "Tensor"
},
"episode_index": {
"dtype": "torch.int64",
"preview": {
"preview": [
0
],
"shape": []
},
"shape": [],
"type": "Tensor"
},
"frame_index": {
"dtype": "torch.int64",
"preview": {
"preview": [
0
],
"shape": []
},
"shape": [],
"type": "Tensor"
},
"index": {
"dtype": "torch.int64",
"preview": {
"preview": [
0
],
"shape": []
},
"shape": [],
"type": "Tensor"
},
"observation.ee_pose": {
"dtype": "torch.float32",
"preview": {
"preview": [
487.6214904785156,
43.20701599121094,
-605.6671142578125,
0.2763311266899109,
-0.3227956295013428
],
"shape": [
7
]
},
"shape": [
7
],
"type": "Tensor"
},
"observation.images.base": {
"dtype": "torch.float32",
"preview": {
"preview": [
0.615686297416687,
0.615686297416687,
0.615686297416687,
0.615686297416687,
0.615686297416687
],
"shape": [
3,
480,
480
]
},
"shape": [
3,
480,
480
],
"type": "Tensor"
},
"observation.images.front": {
"dtype": "torch.float32",
"preview": {
"preview": [
0.5921568870544434,
0.5921568870544434,
0.5882353186607361,
0.5882353186607361,
0.5882353186607361
],
"shape": [
3,
480,
480
]
},
"shape": [
3,
480,
480
],
"type": "Tensor"
},
"observation.images.left": {
"dtype": "torch.float32",
"preview": {
"preview": [
0.5882353186607361,
0.5882353186607361,
0.5882353186607361,
0.5882353186607361,
0.5882353186607361
],
"shape": [
3,
480,
480
]
},
"shape": [
3,
480,
480
],
"type": "Tensor"
},
"observation.images.wrist": {
"dtype": "torch.float32",
"preview": {
"preview": [
0.9215686321258545,
0.9215686321258545,
0.9176470637321472,
0.9176470637321472,
0.9137254953384399
],
"shape": [
3,
480,
480
]
},
"shape": [
3,
480,
480
],
"type": "Tensor"
},
"observation.images.wrist_bottom": {
"dtype": "torch.float32",
"preview": {
"preview": [
0.11372549086809158,
0.11372549086809158,
0.11372549086809158,
0.11372549086809158,
0.11372549086809158
],
"shape": [
3,
480,
480
]
},
"shape": [
3,
480,
480
],
"type": "Tensor"
},
"observation.state": {
"dtype": "torch.float32",
"preview": {
"preview": [
0.026567090302705765,
-0.6289791464805603,
0.03150210902094841,
-2.77478289604187,
0.009506848640739918
],
"shape": [
9
]
},
"shape": [
9
],
"type": "Tensor"
},
"task": {
"preview": "close the cabinet entirely closed",
"type": "str"
},
"task_index": {
"dtype": "torch.int64",
"preview": {
"preview": [
0
],
"shape": []
},
"shape": [],
"type": "Tensor"
},
"timestamp": {
"dtype": "torch.float32",
"preview": {
"preview": [
0.0
],
"shape": []
},
"shape": [],
"type": "Tensor"
}
}
}