Task Visualizations#

We provide a convenience script to write videos for each task’s reset distribution at scripts/get_reset_videos.py for the robosuite tasks we provide in this repository. Set the OUTPUT_FOLDER global variable to the folder where you want to write the videos, and set DATASET_INFOS appropriately if you would like to limit the environments visualized. Then run the script.

The environments are also readily compatible with robosuite visualization scripts such as the demo_random_action.py script and the make_reset_video.py script, but you will need to modify these files to add a import mimicgen line to make sure that robosuite can find these environments.