PiercingDan/capsnet-notebook

An annotated implementation of CapsNet in PyTorch on MNIST heavily based on earlier implementations

MNIST CapsNet

By Danny Luo

Implementation of capsules network model on MNIST in PyTorch (heavily) based on GramAI's implementation by Kenta Iwasaki and subsequent versions. CUDA disabled by default.

This notebook offers further explanation and annotation to offer the user a deep understanding of capsules machinery.

Capsules Paper:

• Sara Sabour, Nicholas Frosst, and Geoffrey E. Hinton Dynamic Routing Between Capsules

References

• https://gist.github.com/kendricktan/9a776ec6322abaaf03cc9befd35508d4
• https://github.com/gram-ai/capsule-networks
• https://github.com/naturomics/CapsNet-Tensorflow