!!! 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

Install PyTorch and dependencies
```
pip install -r requirements.txt
```
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/