A minimal Keras implementation of CapsNet in the paper:

Sara Sabour, Nicholas Frosst, Geoffrey E Hinton. Dynamic Routing Between Capsules. NIPS 2017

Usage

git clone https://github.com/fengwang/minimal-capsule.git
cd ./minimal-capsule
pip install keras tensorflow
python ./mini_capsule.py

Performance

mean absolute error at the first 10 steps:

Acknowledgement

• CapsNet-Keras

• CapsNet-Tensorflow