Top AI Repos — open-source AI, indexed and scored
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Implementation of Single Shot MultiBox Detector in TensorFlow, to detect and classify traffic signs
| Date | Stars |
|---|---|
| 2026-07-24 | 531 |
| 2026-07-25 | 531 |
| 2026-07-28 | 531 |
| 2026-07-30 | 531 |
| 2026-08-06 | 531 |
Today
— stars today
This week
— stars this week
This month
— stars this month
Momentum
0.0
growth rate 0.00%/day
# SSD in TensorFlow: Traffic Sign Detection and Classification ## Overview Implementation of [Single Shot MultiBox Detector (SSD)](https://arxiv.org/abs/1512.02325) in TensorFlow, to detect and classify traffic signs. This implementation was able to achieve 40-45 fps on a GTX 1080 with an Intel Core i7-6700K. *Note this project is still work-in-progress*. The main issue now is model overfitting. I am currently working on pre-training on VOC2012 first, then performing transfer learning over to traffic sign detection. Currently only stop signs and pedestrian crossing signs are detected. Example detection images are below.       The model was trained on the [LISA Traffic Sign Dataset](http://cvrr.ucsd.edu/LISA/lisa-traffic-sign-dataset.html), a dataset of US traffic signs. ## Dependencies * Python 3.5+ * TensorFlow v0.12.0 * Pickle * OpenCV-Python * Matplotlib (optional) ## How to run Clone this repository somewhere, let's refer to it as `$ROOT` Training the model from scratch: * Download the [LISA Traffic Sign Dataset](http://cvrr.ucsd.edu/LISA/lisa-traffic-sign-dataset.html), and store it in a directory `$LISA_DATA` * `cd $LISA_DATA` * Follow instructions in the LISA Traffic Sign Dataset to create 'mergedAnnotations.csv' such that only stop signs and pedestrian crossing signs are shown * `cp $ROOT/data_gathering/create_pickle.py $LISA_DATA` * `python create_pickle.py` * `cd $ROOT` * `ln -s $LISA_DATA/resized_images_* .` * `ln -s $LISA_DATA/data_raw_*.p .` * `python data_prep.py` * This performs box matching between ground-truth boxes and default boxes, and packages the data into a format used later in the pipeline * `python train.py` * This trains the SSD model * `python inference.py -m demo` * This will take the images from sample_images, annotate them, and display them on screen * To run predictions on your own images and/or videos, use the `-i` flag in inference.py (see the code for more details) * Note the model severly overfits at this time ## Differences between original SSD implementation Obivously, we are only detecting certain traffic signs in this implementation, whereas the original SSD implemetation detected a greater number of object classes in the PASCAL VOC and MS COCO datasets. Other notable differences are: * Uses AlexNet as the base network * Input image resolution is 400x260 * Uses a dynamic scaling factor based on the dimensions of the feature map relative to original image dimensions ## Performance As mentioned above, this SSD implementation was able to achieve 40-45 fps on a GTX 1080 with an Intel Core i7 6700K. The inference time is the sum of the neural network inference time, and Non-Maximum Suppression (NMS) time. Overall, the neural network inference time is significantly less than the NMS time, with the neural network inference time generally between 7-8 ms, whereas the NMS time is between 15-16 ms. The NMS algorithm implemented here has not been optimized, and runs on CPU only, so further effort to improve performance can be done there. ## Dataset characteristics The entire LISA Traffic Sign Dataset consists of 47 distinct traffic sign classes. Since we are only concered with a subset of those classes, we only use a subset of the LISA dataset. Also, we ignore all training samples where we do not find a matching default box, further reducing our dataset's size. Due to this process, we end up with very little data to work with. In order to improve on this issue, we can perform image data augmentation, and/or pre-train the model on a larger dataset (e.g. VOC2012, ILSVRC) ## Training process Given the small siz
Excerpt of 4,596 characters
Read on GitHub
11
Would you bet a product on this? Bounded 0–100 and slow moving.
matched fp:76ef5f77606ea437, topic:computer-vision, topic:object-detection
matched fp:76ef5f77606ea437, topic:tensorflow, readme:pre-training