search

yaboo-oyabu / CarND-Capstone

MIT License
2 stars 2 forks source link

A traffic light detection model for real world #6

Closed yaboo-oyabu closed 4 years ago

yaboo-oyabu commented 4 years ago

We may not need to train traffic light detection model for real world because there are many publicly available object detection models that can detect traffic lights. https://github.com/tensorflow/models/tree/master/research/object_detection

yaboo-oyabu commented 4 years ago

@jotredi @CaoFM @xfhg

I tried some pre-trained models listed in TensorFlow model. The following image was captured when I used ssd_mobilenet_v2_coco model: image

Note that I set the lower confidence level (0.2) and display only 'traffic light' class.

jotredi commented 4 years ago

ok great!! yes we can use one of those models for real world images but we still need to predict the traffic light state right?

yaboo-oyabu commented 4 years ago

@jotredi Correct.

yaboo-oyabu commented 4 years ago

Since my precious PR violated one of the requirements for this project, I reverted the above PR. I will dig into this solution in https://stackoverflow.com/questions/53927700/how-to-use-object-detection-api-with-an-old-version-of-tensorflow-v1-3-0 or https://github.com/smasoudn/traffic_light_detection.