现在我们都知道Geoffrey Hinton的胶囊网络(Capsule Network)震动了整个人工智能领域,它将卷积神经网络(CNN)的极限推到一个新的水平。 网上已经有很多的帖子、文章和研究论文在探讨胶囊网络理论,以及它如何做的比传统的CNN更好。因此我不打算介绍这方面的内容,而是尝试使用谷歌的Colaboratory工具在TensorFlow上实现CpNet。
你可以通过下面的几个链接了解CpNet的理论部分:
Geoffrey Hinton的演讲:“卷积神经网络的问题是什么?”胶囊网络正在震动人工智能领域胶囊之间的动态路由现在我们开始写代码。
开始之前,你可以参考我的CoLab Notebook执行以下代码:
CoLab网址:https://goo.gl/43Jvju
现在克隆github上的仓库并安装依赖库。 然后,我们从仓库中取出MNIST数据集,并将其移至父目录:
!git clone https://github.com/bourdakos1/capsule-networks.git
!pip install -r capsule-networks / requirements.txt
!touch capsule-networks / __ init__.py
!mv capsule-networks capsule
!mv capsule / data / ./data/
!ls
现在让我们导入所有的模块:
import os
import tensorflow as tf
from tqdm import tqdm
from capsule.config import cfg
from capsule.uti