preview_dataset / README.md
Emil Ernerfeldt
More general support for any HuggingFace dataset, with streaming
23aef68
|
raw
history blame
892 Bytes
# Rerun visualization of HuggingFace datasets
Tested with LeRobot datasets:
* https://huggingface.co/lerobot
* https://huggingface.co/datasets/lerobot/pusht
https://github.com/rerun-io/python-example-lerobot/assets/1148717/19e9983c-531f-4c48-9b37-37c5cbe1e0bd
## Getting started
Requires Python 3.10 or higher.
```sh
pip install -r requirements.txt
python main.py
```
Example datasets to explore (use `python main.py --dataset`):
* `lerobot/aloha_sim_insertion_human`
* `lerobot/aloha_sim_insertion_scripted`
* `lerobot/aloha_sim_transfer_cube_human`
* `lerobot/aloha_sim_transfer_cube_scripted`
* `lerobot/pusht`
* `lerobot/xarm_lift_medium`
* `nateraw/kitti`
* `sayakpaul/nyu_depth_v2`
## Note for the maintainer
You can update this repository with the latest changes from https://github.com/rerun-io/rerun_template by running `scripts/template_update.py update --languages python`.