huggingface / lerobot

🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning
Apache License 2.0
6.54k stars 582 forks source link

add root argument to the dataset visualizer to visualize local datasets #249

Closed tlpss closed 3 months ago

tlpss commented 3 months ago

What this does

This PR adds the root argument to the dataset visualizer. This parameter is passed to LeRobotDataset.__init__, which enables visualizing local datasets, as an alternative to setting the DATA_DIR env variable.

closes #248

How it was tested

Explain/show how you tested your changes.

How to checkout & try? (for the reviewer)

Examples:

python lerobot/scripts/visualize_dataset.py --repo-id lerobot/pusht --root tests/data/ --episode-index 0 --mode distant

This change is Reviewable

tlpss commented 3 months ago

@Cadene feel free to provide feedback!