/Capsnet

Classifying traffic signs with Capsule Network

Primary LanguagePythonApache License 2.0Apache-2.0

Capsnet

Traffic sign detection with capsules network and using HSV colorspace.

Contributions welcome License completion

Dataset: https://d17h27t6h515a5.cloudfront.net/topher/2017/February/5898cd6f_traffic-signs-data/traffic-signs-data.zip

Requirements

  • Python 3
  • NumPy 1.13
  • Tensorflow 1.4
  • Keras
  • OpenCV 3
  • docopt 0.6.2
  • Sklearn 0.18.1
  • Matplotlib

Train CapNet

$> python train_capsnet.py -h
$> python train_capsnet.py dataset/

Test CapNet

On the Test dataset:

$> python test_capsnet.py outputs/checkpoints/ckpt_name dataset/

On images in from_web folder:

$> python test_web_images.py outputs/checkpoints/ckpt_name

Run detection

$> python main.py