Traffic sign classification on GTSRB dataset using Chainer. The model closely follows this paper.
If you want to get going fast, just do cd tools; source deploy.sh $DATA_PATH
to get everything done. If you need to tweak any of the data processing steps, follow below.
$ python tools/setup_dataset.py -o $DATA_PATH
where $DATA_PATH
is the path where you would like to download the data files to. The directory will be created if it doesn't already exist. You can specify the URL using the optional -i
flag.
Please move into the tools directory before running these scripts.
$ cd tools
$ python oversampler.py
$ python make_annotations.py
$ python generate_means.py
$ python train.py --output /path/to/output --epoch <n> --gpu 0
followed by
$ python detect.py