Zhiyu-Lei / Traffic-Sign-Detection-and-Information-Extraction

Train YOLO object detection model to find traffic signs in the images. Use OCR pipeline to extract the information from the signs with text.

Traffic Sign Detection and Information Extraction

Directory yolov3 stores all the files needed to train the YOLO model using custom data and objects on the GPU server from Google Colab.

To test the YOLO Object Detection and Optical Character Recognition models on other images, open the notebook Test.ipynb using Colab, make sure the test image, yolov3.cfg file and your ".weights" file are all uploaded to Colab, and then run all the cells.