sriramsk/mug_on_platform_20250910_human_heatmapGoal

{
    "codebase_version": "v2.1",
    "robot_type": null,
    "total_episodes": 50,
    "total_frames": 19564,
    "total_tasks": 1,
    "total_videos": 150,
    "total_chunks": 1,
    "chunks_size": 1000,
    "fps": 15,
    "splits": {
        "train": "0:50"
    },
    "data_path": "data/chunk-{episode_chunk:03d}/episode_{episode_index:06d}.parquet",
    "video_path": "videos/chunk-{episode_chunk:03d}/{video_key}/episode_{episode_index:06d}.mp4",
    "features": {
        "action": {
            "dtype": "float32",
            "shape": [
                18
            ],
            "names": [
                "left_waist",
                "left_shoulder",
                "left_shoulder_shadow",
                "left_elbow",
                "left_elbow_shadow",
                "left_forearm_roll",
                "left_wrist_angle",
                "left_wrist_rotate",
                "left_gripper",
                "right_waist",
                "right_shoulder",
                "right_shoulder_shadow",
                "right_elbow",
                "right_elbow_shadow",
                "right_forearm_roll",
                "right_wrist_angle",
                "right_wrist_rotate",
                "right_gripper"
            ]
        },
        "observation.state": {
            "dtype": "float32",
            "shape": [
                18
            ],
            "names": [
                "left_waist",
                "left_shoulder",
                "left_shoulder_shadow",
                "left_elbow",
                "left_elbow_shadow",
                "left_forearm_roll",
                "left_wrist_angle",
                "left_wrist_rotate",
                "left_gripper",
                "right_waist",
                "right_shoulder",
                "right_shoulder_shadow",
                "right_elbow",
                "right_elbow_shadow",
                "right_forearm_roll",
                "right_wrist_angle",
                "right_wrist_rotate",
                "right_gripper"
            ]
        },
        "observation.right_eef_pose": {
            "dtype": "float32",
            "shape": [
                10
            ],
            "names": [
                "rot_6d_0",
                "rot_6d_1",
                "rot_6d_2",
                "rot_6d_3",
                "rot_6d_4",
                "rot_6d_5",
                "trans_0",
                "trans_1",
                "trans_2",
                "gripper_articulation"
            ]
        },
        "action.right_eef_pose": {
            "dtype": "float32",
            "shape": [
                10
            ],
            "names": [
                "rot_6d_0",
                "rot_6d_1",
                "rot_6d_2",
                "rot_6d_3",
                "rot_6d_4",
                "rot_6d_5",
                "trans_0",
                "trans_1",
                "trans_2",
                "gripper_articulation"
            ]
        },
        "observation.images.cam_azure_kinect.transformed_depth": {
            "dtype": "video",
            "shape": [
                720,
                1280,
                1
            ],
            "names": [
                "height",
                "width",
                "channels"
            ],
            "info": "Transformed depth image aligned to color"
        },
        "observation.images.cam_azure_kinect.color": {
            "dtype": "video",
            "shape": [
                720,
                1280,
                3
            ],
            "names": [
                "height",
                "width",
                "channels"
            ],
            "info": "Raw color image"
        },
        "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
        },
        "observation.images.cam_azure_kinect.goal_gripper_proj": {
            "dtype": "video",
            "shape": [
                720,
                1280,
                3
            ],
            "names": [
                "height",
                "width",
                "channels"
            ],
            "info": "Projection of gripper pcd at goal position onto image"
        },
        "observation.points.gripper_pcds": {
            "dtype": "pcd",
            "shape": [
                -1,
                3
            ],
            "names": [
                "N",
                "channels"
            ],
            "info": "Raw gripper point cloud at current position"
        },
        "next_event_idx": {
            "dtype": "int32",
            "shape": [
                1
            ],
            "names": [
                "idx"
            ],
            "info": "Index of next event in the dataset"
        },
        "subgoal": {
            "dtype": "string",
            "shape": [
                1
            ],
            "names": [
                "subgoal"
            ],
            "info": "Caption for subgoal in dataset"
        }
    }
}

[More Information Needed]