Skip to content

chjw1475/Attention-Routing-Capsules

Repository files navigation

Attention-Routing-Capsules

Official implementation of "Attention routing between Capsules"

https://arxiv.org/abs/1907.01750

Updated the arxiv on 2019/08/06 and the code on 2019/08/07.

Tensorflow 1.13.1, Keras 2.2.4

To train a model on the MNIST, CIFAR-10, affNIST, python ar_capsulenet.py or run ar_capsulenet.ipynb

To analyze the pre-trained model, run the summary_mnist_ensembles.ipynb

Accepted as ICCV 2019 workshop paper.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published