This repo contains deep learning model, which classify traffic signs (learned on dataset below) Model is trained to recognize 43 signs.
-
Download dataset from https://d17h27t6h515a5.cloudfront.net/topher/2016/November/581faac4_traffic-signs-data/traffic-signs-data.zip and move data to
datadirectorycreate directory if necessary -
Use predict.py to see predictions
-
To train model use train.py
- image preprocessing: images are normalized, and theirs historgram is equalized.
- load_data uses function to augment data, to create more data to learn model.
- pretrained directory contains trained model.