CapsNet

A PyTorch implementation of CapsNet based on Geoffrey Hinton's paper Dynamic Routing Between Capsules

Requirements

• Anaconda5.0.1
• PyTorch

conda install pytorch torchvision -c soumith
conda install pytorch torchvision cuda80 -c soumith # install it if you have installed cuda

• tqdm

pip install tqdm

Usage

git clone https://github.com/leftthomas/CapsNet.git
cd CapsNet
python main.py

Other Implementations

• capsnet.pytorch:I referred to this repository in my code.

• CapsNet-Keras

• CapsNet-Tensorflow