Glas42/TrafficLightDetectionAI

This is a PyTorch model in the TorchScript format.

This model can classify images of traffic lights into four classes: 0 = red, 1 = yellow, 2 = green, and 3 = not a traffic light. The traffic light images used for training were taken from Euro Truck Simulator 2 and American Truck Simulator.