/Traffic-Signs-Detector

Traffic Signs Detector

Primary LanguageJupyter Notebook

Traffic Signs Detector

To use this repository you will need to follow the requirements listed here and an anaconda installation of python 3.6

To init the project just launch init.sh script:

$ chmod +x init.sh
$ ./init.sh

Since this script downloads the whole GTSDB dataset and initializes darknet submodule, it might take a few minutes

  • Preparing the data can be performed using data-preparation.py script
  • Training and running the network can be performed following AlexeyAB indications
  • You can also check this cool notebook which generates fake traffic signs using a CVAE adapted from TF documentation

References:

This repo is using AlexeyAB repo and YoloV3