The implementation is heavily inspired from the open source contributions of Xifeng Guo.Thank You Guo! For keeping the spirit of open source alive!
This directory contains
-
capsnet ** capsnetv2.py --- Routing with Agreement implementation ** capsulelayers.py-- Capsule layer implementation **capsnetv3.py-- loads saved weights to run tests **tensorflow.sh--runs a shell script for the routing with agreement algo, or loads saved weights
-
convnet ** CNN code
-
class_transition_matrix.py ** dumps confusion matrix, acc, recal