Glas42's picture
Update README.md
4d30778 verified
---
license: gpl-3.0
---
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.