wolfapple/traffic-sign-recognition

Built and trained a deep neural network to classify traffic signs, using PyTorch. The highlights of this solution would be data preprocessing, trained with heavily augmented data and using Spatial Transformer Network.

Jupyter NotebookMIT