Code Monkey home page Code Monkey logo

ampel-pilot-dataset's Introduction

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 [email protected].

Credits

ampel-pilot-dataset's People

Contributors

patvlnta avatar

Stargazers

 avatar  avatar  avatar

Watchers

 avatar

ampel-pilot-dataset's Issues

Model not working as expected in Python

Hi Ampel team -

I hope this message finds you well! I am a student studying machine learning, and I found your project very cool! I'm trying to get a version of your model running in Python.

First, I converted the .weights and .cfg files from this repository into a .h5 file, and took the anchor values from the Ampel-Pilot iOS app here (https://github.com/patVlnta/Ampel-Pilot/blob/c1b2eaaecda6038fa363aa6d12af28153e9f043e/Ampel%20Pilot/utils/Helpers.swift#L15).

I have some yolo code based off of this project (https://github.com/anupamtamrakar/YoloV2/blob/master/Yolov2.ipynb) that simply tries to predict on the sample images in this repository. Unfortunately, it doesn't seem to be working - it is unable to find bounding boxes for most of the images, and draws the bounding boxes in incorrect locations in others. I assume I'm just using the model wrong - can you help point me to what I could be doing wrong? I'd love to be able to get your model working!

Visually, it looks like the boxes are simply skewed off their correct location (they look to be the correct shape), but the skew direction seems to be different each time. Is there some post-processing adjustment I need to make to the bounding boxes?

Thanks for all your help in advance!

Examples:

image (1)

97B81E44-C1F7-4EF9-86B5-3C9AB84961D8

A3A2EB4C-E920-4642-80DF-57712B41CA5A

dataset problem

sorry to bother you, but I encountered difficulty in downloading the dataset that you offered. Can you tell me where can I download this dataset? Thanks a lot!!

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.