| { | |
| "codebase_version": "v3.0", | |
| "robot_type": "franka", | |
| "total_episodes": 1, | |
| "total_frames": 54, | |
| "total_tasks": 1, | |
| "chunks_size": 1000, | |
| "data_files_size_in_mb": 100, | |
| "video_files_size_in_mb": 200, | |
| "fps": 33, | |
| "splits": { | |
| "train": "0:1" | |
| }, | |
| "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": { | |
| "is_first": { | |
| "dtype": "bool", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": null | |
| }, | |
| "is_last": { | |
| "dtype": "bool", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": null | |
| }, | |
| "observation.state.gripper_position": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "gripper" | |
| ] | |
| } | |
| }, | |
| "observation.state.ee_position": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 6 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "x", | |
| "y", | |
| "z", | |
| "roll", | |
| "pitch", | |
| "yaw" | |
| ] | |
| } | |
| }, | |
| "observation.state.joint_position": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 7 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "observation.state": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 8 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6", | |
| "gripper" | |
| ] | |
| } | |
| }, | |
| "observation.state.current_stiffness": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 13 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "observation.state.current_damping": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 13 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "observation.state.current_effort_limit": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 13 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "observation.state.current_torque": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 13 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "observation.state.joint_velocity": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 7 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "observation.state.wrench": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 6 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "fx", | |
| "fy", | |
| "fz", | |
| "tx", | |
| "ty", | |
| "tz" | |
| ] | |
| } | |
| }, | |
| "action.gripper_position": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "gripper" | |
| ] | |
| } | |
| }, | |
| "action.ee_position": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 6 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "x", | |
| "y", | |
| "z", | |
| "roll", | |
| "pitch", | |
| "yaw" | |
| ] | |
| } | |
| }, | |
| "action.joint_position": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 7 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "action.joint_target": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 7 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "action.relative_joint_target": { | |
| "dtype": "float32", | |
| "shape": [ | |
| 7 | |
| ], | |
| "names": { | |
| "axes": [ | |
| "joint_0", | |
| "joint_1", | |
| "joint_2", | |
| "joint_3", | |
| "joint_4", | |
| "joint_5", | |
| "joint_6" | |
| ] | |
| } | |
| }, | |
| "action.high_level_action": { | |
| "dtype": "string", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": null | |
| }, | |
| "action.high_level_args": { | |
| "dtype": "string", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": null | |
| }, | |
| "action.low_level_action": { | |
| "dtype": "string", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": null | |
| }, | |
| "action.low_level_args": { | |
| "dtype": "string", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": null | |
| }, | |
| "is_episode_successful": { | |
| "dtype": "bool", | |
| "shape": [ | |
| 1 | |
| ], | |
| "names": null | |
| }, | |
| "observation.images.front_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": false, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.depth.front_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": true, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.images.gripper_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": false, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.depth.gripper_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": true, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.images.left_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": false, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.depth.left_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": true, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.images.overhead_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": false, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.depth.overhead_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": true, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.images.right_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": false, | |
| "has_audio": false | |
| } | |
| }, | |
| "observation.depth.right_camera": { | |
| "dtype": "video", | |
| "shape": [ | |
| 480, | |
| 640, | |
| 3 | |
| ], | |
| "names": [ | |
| "height", | |
| "width", | |
| "channels" | |
| ], | |
| "info": { | |
| "video.fps": 33.0, | |
| "video.height": 480, | |
| "video.width": 640, | |
| "video.channels": 3, | |
| "video.is_depth_map": true, | |
| "has_audio": false | |
| } | |
| }, | |
| "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 | |
| } | |
| } | |
| } |