Skip to content

drakyanerlanggarizkiwardhana/Traffic-Sign-Detection-YOLO

Repository files navigation

Traffic_sign_detection_YOLO

Detecting traffic signs using YOLO algorithm IMAGE ALT TEXT HERE

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

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 "https://drive.google.com/file/d/171AyNg4zSmz4OXhfcdgU2cxrqTfIV2BD/view?usp=sharing")

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