Skip to content

science-code/capsule-networks

 
 

Repository files navigation

Capsule Networks

A Tensorflow implementation of Capsule Networks

For more information on Capsule Networks please read my Medium article.

This repo was largely based off of naturomics CapsNet-Tensorflow repo with the current differences being an added requirements.txt and the MNIST dataset is included. Power users may want to refer to the naturomics repo for the latest versions.

Usage

Install the requirements:

pip install -r requirements.txt

Start training!

python main.py

Making Inferences

python main.py --is_training False

About

A Tensorflow implementation of Capsule Networks

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.4%
  • R 1.6%