preview_dataset / README.md
jleibs's picture
Switch to docker space type
f242ea8
|
raw
history blame
1.12 kB
metadata
title: Preview Dataset
emoji: 👀
colorFrom: yellow
colorTo: yellow
sdk: docker
app_port: 7860
pinned: false
license: mit

Rerun visualization of HuggingFace datasets

Visualize HuggingFace datasets using Rerun.

Originally built for the LeRobot datasets:

https://github.com/rerun-io/python-example-lerobot/assets/1148717/19e9983c-531f-4c48-9b37-37c5cbe1e0bd

Getting started

Requires Python 3.10 or higher.

pip install -r requirements.txt
python main.py --dataset lerobot/aloha_sim_insertion_human

Example datasets to explore:

  • 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.