| { |
| "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" |
| } |
| } |
| } |