Pedestrian traffic light image dataset (Germany)
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.
- 3696 Images
- 4311 Annotations (63% Red, 37% Green)
The dataset is split into 4 different parts:
A MySql database was used to store image information and image annotations. The dump file is provided in this repository.
Images are stored in the images
database table. Metainformation like file name, file extension and image dimensions can be obtained from here.
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
- Connect to your (local) database server
- Server -> Data Import
- Select the checkbox
Import from Self-Contained File
and provide the path to the SQL dump file - Make sure that
Dump Structure and Data
option is selected on the bottom of the screen - Start Import
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 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.
- pjreddie and all contributors for YOLO/darknet
- Project team @Hochschule Augsburg
- Dataset contribution @University of Tuebingen