Stacked_CapsNet

Stacked CapsNet

CapsNet with multiple layers of Conv.layer Generate capsules from higher-level features with more abstract representations.

Requirement

• torch 1.0.0
• tensorboard 1.12.2
• tensorboardX 1.6
• Pillow 5.4.1

Installation

$ pip install -r requirements.txt

Usage

Download the AffNIST dataset from the website(https://www.cs.toronto.edu/~tijmen/affNIST/) and save it. Clone git.

$ git clone https://github.com/Koko-zep/Stacked_CapsNet.git
$ cd Stacked_CapsNet

Train the model

$ python main.py

Check the results with Tensorboad.

$ cd capsnet_pytorch
$ tensorboard --logdir ./runs

Results

• 3*3Kernel - 3Layers

• EfficientLayer