You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
|`dataset_path`| str | None | If None, it will be automatically downloaded from Hugging Face dataset [URL](https://huggingface.co/datasets/OpenRobotLab/InternBench-M1) on the first run. |
72
-
|`eval_tasks`| list[str]|ALL_EVAL_TASKS| The relative path to the task folder under dataset_path |
72
+
|`eval_tasks`| list[str]|[]| The relative path to the task folder under dataset_path. <br>If `[]`, all tasks in the `dataset_path` will be automatically detected.|
|`gripper_type`| enum | "panda" | End effector selection **when robote_type is `franka`** (`"panda"` or `"robotiq"`) |
77
-
|`franka_camera_enable`| FrankaCameraEnable |`FrankaCameraEnable(realsense=False, `<br>`obs_camera=False, `<br>`obs_camera_2=False)`| Camera activation config:<br>• `realsense`: Ego-view gripper cam<br>• `obs_camera`: Fixed rear-view cam<br>• `obs_camera_2`: Fixed front-view cam <br> Note that this only works **when robote_type is `franka`**|
78
-
|`aloha_split_camera_enable`| AlohaSplitCameraEnable |`AlohaSplitCameraEnable(top_camera=False, `<br>`left_camera=False, `<br>`right_camera=False)`| Camera activation config:<br>• `top_camera`: Ego-view top head cam<br>• `left_camera`: Ego-view left gripper cam<br>• `right_camera`: Ego-view right gripper cam <br> Note that this only works **when robote_type is `aloha_split`**|
77
+
|`franka_camera_enable`| FrankaCameraEnable |`FrankaCameraEnable(`<br>`realsense=False, `<br>`obs_camera=False, `<br>`obs_camera_2=False)`| Camera activation config:<br>• `realsense`: Ego-view gripper cam<br>• `obs_camera`: Fixed rear-view cam<br>• `obs_camera_2`: Fixed front-view cam <br> Note that this only works **when robote_type is `franka`**|
78
+
|`aloha_split_camera_enable`| AlohaSplitCameraEnable |`AlohaSplitCameraEnable(`<br>`top_camera=False, `<br>`left_camera=False, `<br>`right_camera=False)`| Camera activation config:<br>• `top_camera`: Ego-view top head cam<br>• `left_camera`: Ego-view left gripper cam<br>• `right_camera`: Ego-view right gripper cam <br> Note that this only works **when robote_type is `aloha_split`**|
79
79
|`depth_obs`| bool | False | Generate depth maps for active cameras |
80
-
|`max_step`| int |500| Episode termination step threshold |
80
+
|`max_step`| int |1000| Episode termination step threshold |
81
81
|`headless`| bool | True | Disable GUI |
82
82
83
83
@@ -104,8 +104,8 @@ observations: List[Dict] = [
104
104
"robot": {
105
105
"robot_pose": Tuple[array, array], # (position, oritention(quaternion: (w, x, y, z)))
106
106
"joints_state": {
107
-
"positions": array,
108
-
"velocities": array
107
+
"positions": array,# (9,) or (13,) -> panda or robotiq
108
+
"velocities": array# (9,) or (13,) -> panda or robotiq
109
109
},
110
110
"eef_pose": Tuple[array, array], # (position, oritention(quaternion: (w, x, y, z)))
111
111
"sensors": {
@@ -182,8 +182,8 @@ observations: List[Dict] = [
182
182
"robot": {
183
183
"robot_pose": Tuple[array, array], # (position, oritention(quaternion: (w, x, y, z)))
184
184
"joints_state": {
185
-
"positions": array,
186
-
"velocities": array
185
+
"positions": array,# (28,)
186
+
"velocities": array# (28,)
187
187
},
188
188
"left_eef_pose": Tuple[array, array], # (position, oritention(quaternion: (w, x, y, z))) -> left gripper eef pose
189
189
"right_eef_pose": Tuple[array, array], # (position, oritention(quaternion: (w, x, y, z))) -> right gripper eef pose
@@ -216,6 +216,24 @@ observations: List[Dict] = [
216
216
...
217
217
]
218
218
```
219
+
> **Note the following when using `observations["robot"]["joints_state"]["positions"]`.**
0 commit comments