CapsNet-Mxnet

A Mxnet implementation of CapsNet in Hinton's paper Dynamic Routing Between Capsules

Requirements

• Mxnet

Usage

Training

Step 1. Install Mxnet:

$ pip install mxnet

Step 2. Clone this repository with git.

$ git clone https://github.com/AaronLeong/CapsNet_Mxnet.git
$ cd CapsNet_Mxnet

Step 3. Training:

$ python CapsNet.py

Results

Maybe there're some problems in my implementation. Contributions are welcome.

TODO:

• Add reconstruction part, which should be easy.
• Optimize the code implementation and comments

Other Implementations

• CapsNet-Keras: Very good implementation. I referred to this repository in my code.

• CapsNet-Tensorflow

• CapsNet-PyTorch

• capsnet.pytorch