mapaloCat / Traffic-Sign-Detection-System

1 stars 0 forks source link

Traffic-Sign-Detection-System

This repository is about a Computer Vision project that aims at building a traffic sign detection algorithm. The “detector" CNN determines whether a region proposal is a traffic sign or background noise and it classifies it to the corresponding class. In the current implementation, the background noise constitutes an extra class to the 43 classes of the ground truth traffic signs. The presented solution achieves 84.5% mAP (mean Average Precision) in a test set composed of 300 images. The execution time is 738 seconds. The data are available online at the German Traffic Sign Detection Benchmark website.

Here is the link to download the weights for the RPN: https://drive.google.com/open?id=1cin67CB4yR2w1QYxMmwrhF4OiGN0OTyJ

Sample results obtained using the traffic sign detector

84.78% = 1 AP
74.41% = 10 AP
89.63% = 11 AP
89.12% = 12 AP
93.87% = 13 AP
88.18% = 14 AP
100.00% = 15 AP
100.00% = 16 AP
95.00% = 17 AP
63.88% = 18 AP
90.48% = 2 AP
50.00% = 22 AP
100.00% = 23 AP
100.00% = 24 AP
80.00% = 25 AP
29.93% = 26 AP
90.29% = 28 AP
100.00% = 29 AP
63.49% = 3 AP
100.00% = 30 AP
100.00% = 31 AP
100.00% = 32 AP
100.00% = 33 AP
100.00% = 34 AP
100.00% = 35 AP
100.00% = 36 AP
100.00% = 37 AP
77.14% = 38 AP
0.00% = 39 AP
93.46% = 4 AP
100.00% = 40 AP
100.00% = 41 AP
0.00% = 42 AP
73.59% = 5 AP
100.00% = 6 AP
100.00% = 7 AP
94.85% = 8 AP
88.89% = 9 AP
mAP = 84.50%
Traffic sign detection took 738.1926345825195 seconds

Dependencies

Python = 3.6
TensorFlow = 1.15
Keras = 2.1.6