A Tensorflow(v1.4) implementation of Capsule Networks (Dynamic Routing Between Capsules , https://ar...
The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:
I tried to implement the idea in Dynamic Routing Between Capsules
config.pyincludes all hyper parameters that are needed.
utils.pycontains functions regarding loading and saving.
model.pyhas all building blocks for capsNet and whole model implementation.
train.pyis for training.
eval.pyis for evaluation.
$ python train.py
$ python eval.py
| Epoch | 10 | |-----------|--------| | Test Acc. | 99.278 |