esdrascosta / Matrix-Capsules

Matrix Capsules experiment on German Traffic Sign Recognition Benchmark (GTSRB)

!!! NOTES: THIS IS AN ONGOING PROJECT !!!

Matrix Capsules with EM Routing

A PyTorch implementation of Matrix Capsules with EM Routing

German Traffic Sign Recognition Benchmark (GTSRB)

• Dataset GTSRB

Objective

The objective of this project is to use a very good reference Matrix Capsules implementation to experiment on German Traffic Sign Recognition Benchmark (GTSRB)

Usage

1. Install PyTorch and dependencies

pip install -r requirements.txt

2. Start training

python train.py

Test and confusion matrix

python confusion_matrix.py

By defaul it will load _final_gray_model.pth pretrained model

Run Fast Gradient Sign Attack (Adversarial atack)

python adversarial_test.py

Reference

• https://github.com/yl-1993/Matrix-Capsules-EM-PyTorch/