Skip to content

Detect traffic lights and classify the state of them, then give the commands "go" or "stop".

Notifications You must be signed in to change notification settings

JunshengFu/traffic-light-detector

Repository files navigation

Traffic Light Detector

Objective

Detect the traffic lights with TensorFlow Obeject Detection Api, and then use image processing technique to classifer the state of the traffic lights. If the lights are "red" or "yellow", it outputs command "stop"; If the lights are "green", the it outputs "go".

1. How to install Dependence:

2. How to run the code

python main.py
# where commands = [True, False...], and True: go, False: stop

commands = detect_traffic_lights(PATH_TO_TEST_IMAGES_DIR, MODEL_NAME, Num_images, plot_flag=False)

4. An example

A very simple network is used here:

Output command:
stop

About

Detect traffic lights and classify the state of them, then give the commands "go" or "stop".

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages