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.
Install the requirements:
pip install -r requirements.txt
Start training!
python main.py
python main.py --is_training False