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".

• environment-gpu.yml environment file with GPU

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)

A very simple network is used here:

stop