Simple traffic light classifier to be integrated in the capstone project.
Number of frames: 1324 frames
Resolution: 800x600 pixel
Classes: {none, red, yellow, green}
Number of frames: 1151 frames
Resolution: 800x600 pixel
Classes: {none, red, green}
The network has been from the dataset obtained merging the two dataset above.
All three datasets are available in this shared folder.
train.py
: entry point for training the modeltest.py
: entry point for testing the model from pretrained weightstraffic_light_dataset.py
: containsTrafficLightDataset
class to abstract over the raw datatraffic_light_classifier.py
: containsTrafficLightClassifier
deep network for classification