CapsLayer: An advanced library for capsule theory
Capsule theory is a potential research proposed by Geoffrey E. Hinton et al, where he describes the shortcomings of the Convolutional Neural Networks and how Capsules could potentially circumvent these problems such as "pixel attack" and create more robust Neural Network Architecture based on Capsules Layer.
We expect that this theory will definitely contribute to Deep Learning Industry and we are excited about it. For the same reason we are proud to introduce CapsLayer, an advanced library for the Capsule Theory, integrating capsule-relevant technologies, providing relevant analysis tools, developing related application examples, and probably most important thing: promoting the development of capsule theory.
This library is based on Tensorflow and has a similar API with it but designed for capsule layers/models.
Features
-
TensorFlow-like API for building Neural Nets block, see API docs for more details:
- capslayer.layers.conv2d
- capslayer.layers.conv1d
- capslayer.layers.fully_connected/dense
- capslayer.layers.primaryCaps
- capslayer.losses.spread_loss
- capslayer.losses.margin_loss
-
Datasets support:
-
Capsule Nets Model examples:
-
Algorithm support:
- Routing-by-agreement: including EM Routing and Dynamic Routing
If you want us to support more features, let us know by opening Issues or sending E-mail to naturomics.liao@gmail.com
Documentation
- Installation
- Tutorials for running CapsNet on supported dataset (MNIST/CIFAR10 etc.) or your own dataset, or building your network with Capsule Layer
- Theoretical Analysis
Contributions
Feel free to send your pull request or open issues
Citation
If you find it is useful, please cite our project by the following BibTex entry:
@misc{HuadongLiao2017,
title = {CapsLayer: An advanced library for capsule theory},
author = {Huadong Liao, Jiawei He},
year = {2017}
publisher = {GitHub},
journal = {GitHub Project},
howpublished = {\url{http://naturomics.com/CapsLayer}},
}
Note:
We are considering to write a paper for this project, but before that, please cite the above Bibtex entry if you find it helps.
License
Apache 2.0 license.