task_categories:reinforcement-learning task_categories:robotics license:mit size_categories:10K<n<100K format:arrow modality:image modality:timeseries library:datasets library:mlcroissant region:us
---
task_categories:
- reinforcement-learning
- robotics
license: mit
---

# Ball Maze Environment Dataset

This dataset contains episodes of a ball maze environment, converted to the LeRobot format for visualization and training.

## Dataset Structure

```
Sagar18/ball-maze-lerobot/
├── data/                 # Main dataset files
├── metadata/            # Dataset metadata
│   └── info.json       # Configuration and version info
├── episode_data_index.safetensors  # Episode indexing information
└── stats.safetensors    # Dataset statistics
```

## Features

- RGB camera view of the maze environment
- State space representation (x, y, dx, dy)
- Actions (2D force vector)
- Episode indexing and timestamps

## Usage

```python
from datasets import load_dataset

dataset = load_dataset("Sagar18/ball-maze-lerobot", "data")  # Use the data subfolder
```

## Metadata

The metadata contains configuration information and can be accessed separately:

```python
import json
from huggingface_hub import hf_hub_download

info_path = hf_hub_download(repo_id="Sagar18/ball-maze-lerobot", filename="metadata/info.json")
with open(info_path) as f:
    info = json.load(f)
```
