Hello. This is a project that using deep learning to identify traffic light signals. It is built using fast.ai. A resnet with 34 layers was trained with images from Bing's Image Search API. Refer to Traffic_Light_Detector.py for the code used in the Jupyter Notebook to actually train the model. To actually run the Detector, refer to Traffic_Detector.ipynb. Run that Notebook, and make sure you have the .pkl file downloaded. By running the second box, you may upload your own images and have them classified.
andwu03/Traffic_Light_Detector
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published