Home

Awesome

CapsNet-Mxnet

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

Requirements

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:

Other Implementations