Visualize

Note

Ongoing support for the LeRobot fork is deprecated. We recommend using the plugin-based LeRobot integration instead.

However, this integration is necessary for users who wish to use the openpi integration.

Remote Visualization

If you uploaded your dataset to the Hugging Face Hub using --control.push_to_hub=true, you can visualize your dataset online. To do so, copy and paste your repository ID into the provided field. Your repository ID follows the format:

<huggingface-username>/<dataset-id>

You can retrieve your repository ID by running the following command in your terminal:

echo ${HF_USER}/trossen_ai_stationary_test

echo ${HF_USER}/trossen_ai_mobile_test

echo ${HF_USER}/trossen_ai_solo_test

Here, ${HF_USER} represents your Hugging Face username, and trossen_ai_xxxxxxx_test is the dataset ID you provided during the upload.

Local Visualization

If you didn’t upload the dataset (i.e., you used --control.push_to_hub=false), you can still visualize it locally with:

python lerobot/scripts/visualize_dataset_html.py \
    --repo-id <local_dir_name>/trossen_ai_stationary_test \
    --load-from-hf-hub 0

python lerobot/scripts/visualize_dataset_html.py \
    --repo-id <local_dir_name>/trossen_ai_mobile_test \
    --load-from-hf-hub 0

python lerobot/scripts/visualize_dataset_html.py \
    --repo-id <local_dir_name>/trossen_ai_solo_test \
    --load-from-hf-hub 0