Capsule Networks ("Dynamic Routing between Capsules" by Geoff Hinton et al.)
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
CONTRIBUTING.md
CapsNet.py
LICENSE
README.md
capsnet_layer.py
config.py
download_dataset.py
main.py
utils.py

README.md

CapsNet

Contributions welcome License

This is a tensorflow implementation of Capsule Networks ("Dynamic Routing between Capsules" by Geoff Hinton et al.).

This repository is under construction.

Requirements

Cloning the repository

$ https://github.com/prajjwal1/capsnet

$ cd capsnet

Downloading the data

For MNIST

$ python download_data.py

For Fashion MNIST

$ python download_data.py --dataset fashion-mnist --save_to data/fashion-mnist

Training the model

For MNIST

$ python main.py

For Fashion MNIST

$ python main.py --dataset fashion-mnist

Datasets

  • MNIST
  • Fashion MNIST
  • CIFAR 10

Configuration

Default parameters are defined in config.py . Batch size has been set to 128 and epoch 150. You may need to modify those parameters as per your application.

Contribution

Contributions are always welcome. See CONTRIBUTING.MD for more details.