home capsule_networks
View code on Github
这是胶囊间动态路由的 PyTorch 实现/教程。
Capsule 网络是一种神经网络架构,它以胶囊的形式嵌入特征,并通过投票机制将它们路由到下一层胶囊。
与其他模型实现不同,我们提供了一个示例,因为仅使用模块很难理解某些概念。这是使用胶囊对 MNIST 数据集进行分类的模型的带注释的代码
该文件包含了 Capsule Networks 核心模块的实现。
我用 jindongwang/pytorch-CapsuleNet 来澄清我对这篇论文的一些困惑。
这是一本在 MNIST 数据集上训练 Capsule 网络的笔记本。