UMI Visualizer

A high-performance visualizer for Universal Manipulation Interface (UMI) datasets using Rerun.

Installation

1. Generate your "dataset.zarr" data from the UMI repository (Reference UMI repository)

This project uses uv for dependency management.

2. Install uv (for detailed installation guide refer Install uv):

   curl -sSf [https://astral.sh/uv/install.sh](https://astral.sh/uv/install.sh) | sh
   

3. Install System Dependencies (Linux only):

   sudo apt-get update && sudo apt-get install -y libblosc-dev liblz4-dev
   

4. Run the Visualizer:

   uv run main.py /path/to/your/dataset.zarr --episode 1
   

5. Use [SPACEBAR] to start or pause the Visualization

6. Close the Rerun Window and press Ctrl+C in the terminal to stop the execution

7. Running tests:

   uv run black .
   uv run ruff check . --fix
   uv run mypy main.py
   uv run -m pytest

---