size_categories:10B<n<100B language:en license:mit Autonomous Driving CARLA Simulator ImitationLearning region:us
---
license: mit
language:
- en
tags:
- Autonomous Driving
- CARLA Simulator
- ImitationLearning
size_categories:
- 10B<n<100B
pretty_name: S
---

This dataset consists of 80 episodes of driving data collected using an autopilot agent in CARLA simulator for training imitation learning models for autonomous driving tasks.
Each frame is structured as follows:
```
frame_data = {
            'frame': the frame index,
            'hlc': an integer representing the high-level command,
            'light': an integer representing current traffic light status,
            'controls': an array of [throttle, steer, brake],
            'measurements': current speed in km/h,
            'rgb': rgb camera image,
            'segmentation': ground truth segmentation image,
        }
```

This dataset is used in [this project](https://github.com/TheRoboticsClub/gsoc2023-Meiqi_Zhao) and the trained models are available [here](https://huggingface.co/nightmare-nectarine/segmentation-based-imitation-learning-in-CARLA). Check out the [example code](https://github.com/TheRoboticsClub/gsoc2023-Meiqi_Zhao/blob/main/src/ModifiedDeepestLSTMTinyPilotNet/utils/load_dataset.py) for loading the dataset.