CapsNet-tensorflow-jupyter A Tensorflow implementation of CapsNet , based on my understanding. This repository is built with an aim to simplify the concept, implement and understand it.
Any suggestions, or any mistake you find, I'll be happy to entertain.
Credits for this amazing comparison: https://github.com/naturomics/CapsNet-Tensorflow
Requirements python3
numpy==1.13.1
tensorflow==1.3.0
UsageClone this repository with git
.
$ git clone https://github.com/rrqq/CapsNet-tensorflow-jupyter.git
$ cd CapsNet-tensorflow-jupyter-master Run OptionsUse Jupyter Notebook (with visualizations)
From linux terminal:
$ script -c "sudo python3 main.py" train_log.txt For Dynamic Loop implementation:
$ script -c "sudo python3 dynamic_main.py" train_log_dynamic.txt Test Accuracy (dynamic_main.py): 99.38 % First round with default parameters as in paper. The test set was the default provided by Tensorflow.
ReferenceIn depth analysis.