CoolFace
Datasetpublic

Pi-robot/place_blue_bottle_split

This dataset was created using LeRobot. Dataset Structure meta/info.json: { "codebase_version": "v2.1", "robot_type": "a2d", "total_episodes": 325, "total_frames": 183325, "total_tasks": 4, "total_videos": 975, "total_chunks": 1, "chunks_size": 1000, "fps": 25, "splits": { "train": "0:325" }, "data_path": "data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet", "video_path":… See the full description on the dataset page: https://huggingface.co/datasets/Pi-robot/place_blue_bottle_split.

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
0likes190downloads
README.md259 linesDownload Raw Back to root
1---2license: apache-2.03task_categories:4- robotics5tags:6- LeRobot7configs:8- config_name: default9  data_files: data/*/*.parquet10---11 12This dataset was created using [LeRobot](https://github.com/huggingface/lerobot).13 14## Dataset Description15 16 17 18- **Homepage:** [More Information Needed]19- **Paper:** [More Information Needed]20- **License:** apache-2.021 22## Dataset Structure23 24[meta/info.json](meta/info.json):25```json26{27    "codebase_version": "v2.1",28    "robot_type": "a2d",29    "total_episodes": 325,30    "total_frames": 183325,31    "total_tasks": 4,32    "total_videos": 975,33    "total_chunks": 1,34    "chunks_size": 1000,35    "fps": 25,36    "splits": {37        "train": "0:325"38    },39    "data_path": "data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet",40    "video_path": "videos/chunk-{episode_chunk:03d}/{video_key}/episode_{episode_index:06d}.mp4",41    "features": {42        "action": {43            "dtype": "float32",44            "shape": [45                1646            ],47            "names": [48                "left_shoulder_roll",49                "left_shoulder_pitch",50                "left_shoulder_yaw",51                "left_elbow_pitch",52                "left_wrist_roll",53                "left_wrist_pitch",54                "left_wrist_yaw",55                "right_shoulder_roll",56                "right_shoulder_pitch",57                "right_shoulder_yaw",58                "right_elbow_pitch",59                "right_wrist_roll",60                "right_wrist_pitch",61                "right_wrist_yaw",62                "left_gripper",63                "right_gripper"64            ]65        },66        "observation.state": {67            "dtype": "float32",68            "shape": [69                1670            ],71            "names": [72                "left_shoulder_roll",73                "left_shoulder_pitch",74                "left_shoulder_yaw",75                "left_elbow_pitch",76                "left_wrist_roll",77                "left_wrist_pitch",78                "left_wrist_yaw",79                "right_shoulder_roll",80                "right_shoulder_pitch",81                "right_shoulder_yaw",82                "right_elbow_pitch",83                "right_wrist_roll",84                "right_wrist_pitch",85                "right_wrist_yaw",86                "left_gripper",87                "right_gripper"88            ]89        },90        "observation.force": {91            "dtype": "float32",92            "shape": [93                1294            ],95            "names": [96                "left_force_x",97                "left_force_y",98                "left_force_z",99                "left_torque_x",100                "left_torque_y",101                "left_torque_z",102                "right_force_x",103                "right_force_y",104                "right_force_z",105                "right_torque_x",106                "right_torque_y",107                "right_torque_z"108            ]109        },110        "observation.ee_pose": {111            "dtype": "float32",112            "shape": [113                12114            ],115            "names": [116                "left_ee_x",117                "left_ee_y",118                "left_ee_z",119                "left_ee_rx",120                "left_ee_ry",121                "left_ee_rz",122                "right_ee_x",123                "right_ee_y",124                "right_ee_z",125                "right_ee_rx",126                "right_ee_ry",127                "right_ee_rz"128            ]129        },130        "observation.images.head_color": {131            "shape": [132                480,133                640,134                3135            ],136            "names": [137                "height",138                "width",139                "channels"140            ],141            "info": {142                "video.fps": 25.0,143                "video.height": 480,144                "video.width": 640,145                "video.channels": 3,146                "video.codec": "h264",147                "video.pix_fmt": "yuv420p",148                "video.is_depth_map": false,149                "has_audio": false150            },151            "dtype": "video"152        },153        "observation.images.head_depth": {154            "shape": [155                480,156                640,157                1158            ],159            "names": [160                "height",161                "width",162                "channels"163            ],164            "info": null,165            "dtype": "image"166        },167        "observation.images.hand_left_color": {168            "shape": [169                480,170                640,171                3172            ],173            "names": [174                "height",175                "width",176                "channels"177            ],178            "info": {179                "video.fps": 25.0,180                "video.height": 480,181                "video.width": 640,182                "video.channels": 3,183                "video.codec": "h264",184                "video.pix_fmt": "yuv420p",185                "video.is_depth_map": false,186                "has_audio": false187            },188            "dtype": "video"189        },190        "observation.images.hand_right_color": {191            "shape": [192                480,193                640,194                3195            ],196            "names": [197                "height",198                "width",199                "channels"200            ],201            "info": {202                "video.fps": 25.0,203                "video.height": 480,204                "video.width": 640,205                "video.channels": 3,206                "video.codec": "h264",207                "video.pix_fmt": "yuv420p",208                "video.is_depth_map": false,209                "has_audio": false210            },211            "dtype": "video"212        },213        "timestamp": {214            "dtype": "float32",215            "shape": [216                1217            ],218            "names": null219        },220        "frame_index": {221            "dtype": "int64",222            "shape": [223                1224            ],225            "names": null226        },227        "episode_index": {228            "dtype": "int64",229            "shape": [230                1231            ],232            "names": null233        },234        "index": {235            "dtype": "int64",236            "shape": [237                1238            ],239            "names": null240        },241        "task_index": {242            "dtype": "int64",243            "shape": [244                1245            ],246            "names": null247        }248    }249}250```251 252 253## Citation254 255**BibTeX:**256 257```bibtex258[More Information Needed]259```