/capsules-utils

Fully vectorized and interpretable Pytorch implementation of capsule layers (and CapsNet) as described in the NIPS 2017 paper 'Dynamic Routing Between Capsules' by S. Sabour, N. Frosst and G.E. Hinton.

Primary LanguagePythonMIT LicenseMIT

Stargazers