JunshengFu / traffic-light-detector

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

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:

• environment-gpu.yml environment file with GPU

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