Skip to content

Commit d0b4d9d

Browse files
author
Yu Fang
committed
Add multiple camera example with collect_human_demonstrations in robosuite
1 parent e071051 commit d0b4d9d

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

docs/algorithms/demonstrations.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ We provide teleoperation utilities that allow users to control the robots with i
77
- `directory:` path to a folder for where to store the pickle file of collected demonstrations
88
- `environment:` name of the environment you would like to collect the demonstrations for
99
- `device:` either "keyboard" or "spacemouse" or "dualsense" or "mjgui"
10+
- `renderer:` Mujoco's builtin interactive viewer (mjviewer) or OpenCV viewer (mujoco)
11+
- `camera:` Pass multiple camera names to enable multiple views. Note that the "mujoco" renderer must be enabled when using multiple views, while "mjviewer" is not supported.
1012

1113
See the [devices page](https://robosuite.ai/docs/modules/devices.html) for details on how to use the devices.
1214

robosuite/scripts/collect_human_demonstrations.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,9 +237,10 @@ def gather_demonstrations_as_hdf5(directory, out_dir, env_info):
237237
)
238238
parser.add_argument(
239239
"--camera",
240+
nargs="*",
240241
type=str,
241242
default="agentview",
242-
help="Which camera to use for collecting demos",
243+
help="List of camera names to use for collecting demos. Pass multiple names to enable multiple views. Note: the `mujoco` renderer must be enabled when using multiple views; `mjviewer` is not supported."
243244
)
244245
parser.add_argument(
245246
"--controller",

0 commit comments

Comments
 (0)