Metadata-Version: 2.4
Name: domin
Version: 0.1.0
Summary: Easy and performant synthetic dataset generation for VLA models using Isaac Sim multi-env simulation.
License-Expression: Apache-2.0
Project-URL: Homepage, https://github.com/nimit/domin
Project-URL: Issues, https://github.com/nimit/domin/issues
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
License-File: NOTICE
Requires-Dist: torch>=2.4.0
Requires-Dist: torchvision>=0.19.0
Requires-Dist: datasets>=4.4.1
Requires-Dist: jsonlines>=4.0.0
Requires-Dist: av>=16.0.1
Requires-Dist: decord>=0.6.0
Requires-Dist: deepdiff>=8.6.1
Requires-Dist: isaaclab[isaacsim]>=0.48.8
Requires-Dist: stable-baselines3>=2.7.0
Requires-Dist: scene-synthesizer
Requires-Dist: xatlas
Provides-Extra: dev
Requires-Dist: pytest; extra == "dev"
Requires-Dist: ruff; extra == "dev"
Requires-Dist: isort; extra == "dev"
Dynamic: license-file

# Domin

> _A high-throughput orchestrator for synthetic robotics datasets; named after Harry Domin of R.U.R., the visionary who sought to free humanity from labor through the mass production of robots._

`domin` makes the generation of synthetic datasets for Visual-Language-Action (VLA) models easy, performant, and scalable.

While most current simulation in robotics is geared towards Reinforcement Learning (RL), this package leverages the same massively parallel, multi-environment paradigm to generate training data. This allows you to produce large-scale datasets using [NVIDIA Isaac Sim](https://developer.nvidia.com/isaac-sim) and [Isaac Lab](https://github.com/isaac-sim/IsaacLab) with a **single configuration file**.

## Features

This package generates datasets in the popular **LeRobot** format, compatible with Hugging Face's [LeRobot](https://github.com/huggingface/lerobot) library. The core dataset builder is a modified and extended version of the LeRobot dataset builder, optimized for simulation environments.

Key features inherited from our extended `dataset_builder`:

- **Simultaneous Episode Recording**: Record multiple episodes (environments) in parallel for high throughput, significantly speeding up data generation.
- **Story Mode**: Episodes are grouped into "stories" (batches) for efficient management and synchronized resetting.
- **Scheduled Re-recording**: Robust handling of failed episodes. If an episode fails, it is automatically cleared and scheduled for a retry in the next batch, ensuring dataset completeness without manual intervention.
- **Metadata & Custom Metrics**: Easily save arbitrary metadata (e.g., success rates, simulation parameters) and automatically compute episode statistics in the dataset's `info.json`.
- **LeRobot Format Compatibility**: Produces datasets in the standard LeRobot format (Parquet files with embedded or external images), ready for training.

## Installation

You can install this package via pip. Note that you must specify the NVIDIA PyPI index for `isaacsim` and related dependencies.

```bash
pip install . --extra-index-url https://pypi.nvidia.com
```

For advanced installation instructions, including setting up Isaac Sim and creating a development environment, please see [CONTRIBUTING.md](CONTRIBUTING.md).

## Usage

To generate a dataset, use the `domin-gen` command with a dataset configuration file.

The dataset configuration file should inherit `domin.base_dataset_config.BaseDatasetConfig` (see example).

### Example

```bash
domin-gen examples/dexterous_dataset_config.py --num_envs 10 --num_episodes 100
```

### Arguments

- `config_path`: Path to the Python file containing the dataset configuration (e.g., `examples/dexterous_dataset_config.py`).
- `--num_envs`: (Optional) Number of parallel environments to simulate (default: 1).
- `--num_episodes`: (Required) Total number of episodes to record.

## Acknowledgements

This project builds upon the excellent work of the **Hugging Face LeRobot** team. The `domin.dataset_builder` module is a modified adaptation of their dataset building tools, tailored for the specific needs of massive parallel simulation in Isaac Lab. We gratefully acknowledge their contributions to the open-source robotics community.
