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