Skip to content

Traffic sign classification on GTSRB dataset using Chainer

Notifications You must be signed in to change notification settings

rcalland/chainer-GTSRB

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

chainer-GTSRB

Traffic sign classification on GTSRB dataset using Chainer. The model closely follows this paper.

Setup

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.

Download data

$ 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.

Preprocess data

Please move into the tools directory before running these scripts.

$ cd tools
$ python oversampler.py
$ python make_annotations.py
$ python generate_means.py

Training & Detection

$ python train.py --output /path/to/output --epoch <n> --gpu 0

followed by

$ python detect.py

About

Traffic sign classification on GTSRB dataset using Chainer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages