Traffic_sign_detection_YOLO
Detecting traffic signs using YOLO algorithm
Clone the repository
git clone https://github.com/AmeyaWagh/Traffic_sign_detection_YOLO.git
Goto darkflow and build cython extension by running
cd darkflow
python3 setup.py build_ext --inplace
Then build globally with
pip install .
Check if "flow" works with "flow --h"
flow --h
Go back and create a new folder called "dataset" in base directory. Download and extract LISA dataset into the dataset folder
cd ..
mkdir dataset
run datasetGenerator.py
python3 datasetGenerator.py
https://drive.google.com/file/d/171AyNg4zSmz4OXhfcdgU2cxrqTfIV2BD/view?usp=sharing")
goto darkflow and create "built_graph" directory inside darkflow if you are not training, and save pb and meta files there (pb and meta files can be downloaded here "cd darkflow
mkdir built_graph
set GPU to 0.0 in the config3.json if not using GPU
{
"yoloConfig":{
"pbLoad": "./built_graph/tiny-yolo-voc27.pb",
"metaLoad": "./built_graph/tiny-yolo-voc27.meta",
"labels":"../labels.txt",
"threshold":0.01,
"gpu":0.7
},
"dataset":"./dataset"
}
Run YOLO
./runYOLO
Training
cd darkflow
./trainYOLO