https://github.com/piercingdan/capsnet-notebook
An annotated implementation of CapsNet in PyTorch on MNIST heavily based on earlier implementations
https://github.com/piercingdan/capsnet-notebook
Last synced: 11 months ago
JSON representation
An annotated implementation of CapsNet in PyTorch on MNIST heavily based on earlier implementations
- Host: GitHub
- URL: https://github.com/piercingdan/capsnet-notebook
- Owner: PiercingDan
- Created: 2018-06-04T12:05:02.000Z (about 8 years ago)
- Default Branch: master
- Last Pushed: 2018-06-04T12:08:03.000Z (about 8 years ago)
- Last Synced: 2025-02-14T18:23:05.226Z (over 1 year ago)
- Language: Jupyter Notebook
- Size: 3.91 KB
- Stars: 1
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 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