胶囊网络

这是胶囊间动态路由PyTorch 实现/教程。

Capsule 网络是一种神经网络架构,它以胶囊的形式嵌入特征,并通过投票机制将它们路由到下一层胶囊。

与其他模型实现不同,我们提供了一个示例,因为仅使用模块很难理解某些概念。这是使用胶囊对 MNIST 数据集进行分类的模型的带注释的代码

该文件包含了 Capsule Networks 核心模块的实现。

我用 jindongwang/pytorch-CapsuleNet 来澄清我对这篇论文的一些困惑。

这是一本在 MNIST 数据集上训练 Capsule 网络的笔记本。

Open In Colab