/Ampel-Pilot-Dataset

Pedestrian traffic light image dataset (Germany)

MIT LicenseMIT

Ampel-Pilot-Dataset

Pedestrian traffic light image dataset (Germany)

Preview

The dataset has been collected in a joint effort between the Hochschule Augsburg and the University of Tuebingen. Students were able to contribute by sending their pictures of pedestrian traffic lights with the LightsCatcher application.

The dataset has been used to train an object detection algorithm which is used in the Ampel-Pilot mobile application for visually impaired mobile users. It can be used as guidance for determining the current phase of a pedestrian traffic light.

Stats

  • 3696 Images
  • 4311 Annotations (63% Red, 37% Green)

Download

The dataset is split into 4 different parts:

Database

A MySql database was used to store image information and image annotations. The dump file is provided in this repository.

Images

Images are stored in the images database table. Metainformation like file name, file extension and image dimensions can be obtained from here.

Annotations

Annotations are stored in the objects table. Information like related image, class label (Red, Green) and position of the bounding box within the image can be obtained from here.

! IMPORTANT !

  • The bounding box dimensions are kept relative to the image dimensions.
  • x,y is the center of the bounding box

Installation (MySQL Workbench)

  1. Connect to your (local) database server
  2. Server -> Data Import
  3. Select the checkbox Import from Self-Contained File and provide the path to the SQL dump file
  4. Make sure that Dump Structure and Data option is selected on the bottom of the screen
  5. Start Import

Model

A YOLOv2 object detection model (tiny) was trained based on this dataset using 3062 Images for training and 630 Images for validation.

Light Phase Recall Precision IoU
Red 0.796 0.739 0.602
Green 0.734 0.688 0.601

The model including the config file can be found here.

Contributions

Contributions to the dataset are always very welcome. If you have any further questions, ideas or enquiries, feel free to get in contact either by opening an issue or email valpaet@gmail.com.

Credits