runyud's picture
[add] add G1 loco-manip sim dataset
6e3172c verified
{
"codebase_version": "v2.1",
"robot_type": "humanoid",
"total_episodes": 103,
"total_frames": 78369,
"total_tasks": 2,
"total_videos": 103,
"total_chunks": 1,
"chunks_size": 1000,
"fps": 50.0,
"splits": {
"train": "0:100"
},
"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": {
"observation.images.ego_view": {
"dtype": "video",
"shape": [
480,
640,
3
],
"names": [
"height",
"width",
"channel"
],
"video_info": {
"video.fps": 50.0,
"video.codec": "h264",
"video.pix_fmt": "yuv420p",
"video.is_depth_map": false,
"has_audio": false
}
},
"observation.state": {
"dtype": "float64",
"shape": [
43
],
"names": [
"left_hip_pitch_joint",
"left_hip_roll_joint",
"left_hip_yaw_joint",
"left_knee_joint",
"left_ankle_pitch_joint",
"left_ankle_roll_joint",
"right_hip_pitch_joint",
"right_hip_roll_joint",
"right_hip_yaw_joint",
"right_knee_joint",
"right_ankle_pitch_joint",
"right_ankle_roll_joint",
"waist_yaw_joint",
"waist_roll_joint",
"waist_pitch_joint",
"left_shoulder_pitch_joint",
"left_shoulder_roll_joint",
"left_shoulder_yaw_joint",
"left_elbow_joint",
"left_wrist_roll_joint",
"left_wrist_pitch_joint",
"left_wrist_yaw_joint",
"left_hand_index_0_joint",
"left_hand_index_1_joint",
"left_hand_middle_0_joint",
"left_hand_middle_1_joint",
"left_hand_thumb_0_joint",
"left_hand_thumb_1_joint",
"left_hand_thumb_2_joint",
"right_shoulder_pitch_joint",
"right_shoulder_roll_joint",
"right_shoulder_yaw_joint",
"right_elbow_joint",
"right_wrist_roll_joint",
"right_wrist_pitch_joint",
"right_wrist_yaw_joint",
"right_hand_index_0_joint",
"right_hand_index_1_joint",
"right_hand_middle_0_joint",
"right_hand_middle_1_joint",
"right_hand_thumb_0_joint",
"right_hand_thumb_1_joint",
"right_hand_thumb_2_joint"
]
},
"observation.eef_state": {
"dtype": "float64",
"shape": [
14
],
"names": [
"left_wrist_pos",
"left_wrist_abs_quat",
"right_wrist_pos",
"right_wrist_abs_quat"
]
},
"action": {
"dtype": "float64",
"shape": [
43
],
"names": [
"left_hip_pitch_joint",
"left_hip_roll_joint",
"left_hip_yaw_joint",
"left_knee_joint",
"left_ankle_pitch_joint",
"left_ankle_roll_joint",
"right_hip_pitch_joint",
"right_hip_roll_joint",
"right_hip_yaw_joint",
"right_knee_joint",
"right_ankle_pitch_joint",
"right_ankle_roll_joint",
"waist_yaw_joint",
"waist_roll_joint",
"waist_pitch_joint",
"left_shoulder_pitch_joint",
"left_shoulder_roll_joint",
"left_shoulder_yaw_joint",
"left_elbow_joint",
"left_wrist_roll_joint",
"left_wrist_pitch_joint",
"left_wrist_yaw_joint",
"left_hand_index_0_joint",
"left_hand_index_1_joint",
"left_hand_middle_0_joint",
"left_hand_middle_1_joint",
"left_hand_thumb_0_joint",
"left_hand_thumb_1_joint",
"left_hand_thumb_2_joint",
"right_shoulder_pitch_joint",
"right_shoulder_roll_joint",
"right_shoulder_yaw_joint",
"right_elbow_joint",
"right_wrist_roll_joint",
"right_wrist_pitch_joint",
"right_wrist_yaw_joint",
"right_hand_index_0_joint",
"right_hand_index_1_joint",
"right_hand_middle_0_joint",
"right_hand_middle_1_joint",
"right_hand_thumb_0_joint",
"right_hand_thumb_1_joint",
"right_hand_thumb_2_joint"
]
},
"action.eef": {
"dtype": "float64",
"shape": [
14
],
"names": [
"left_wrist_pos",
"left_wrist_abs_quat",
"right_wrist_pos",
"right_wrist_abs_quat"
]
},
"teleop.navigate_command": {
"dtype": "float64",
"shape": [
3
],
"names": [
"lin_vel_x",
"lin_vel_y",
"ang_vel_z"
]
},
"teleop.base_height_command": {
"dtype": "float64",
"shape": [
1
],
"names": "base_height_command"
},
"observation.sim.seed": {
"dtype": "int32",
"shape": [
1
]
},
"observation.sim.max_mujoco_state_len": {
"dtype": "int32",
"shape": [
1
]
},
"observation.sim.mujoco_state_len": {
"dtype": "int32",
"shape": [
1
]
},
"observation.sim.mujoco_state": {
"dtype": "float64",
"shape": [
800
]
},
"observation.sim.left_wrist": {
"dtype": "float64",
"shape": [
16
]
},
"observation.sim.right_wrist": {
"dtype": "float64",
"shape": [
16
]
},
"observation.sim.left_fingers": {
"dtype": "float64",
"shape": [
400
]
},
"observation.sim.right_fingers": {
"dtype": "float64",
"shape": [
400
]
},
"observation.sim.target_upper_body_pose": {
"dtype": "float64",
"shape": [
31
]
},
"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
}
}
}