MatrixCapsulesWithEMRouting:在TensorFlow中实现``具有EM路由的矩阵胶囊''

具有EM路由的矩阵胶囊 TensorFlow实现。 要求 使用以下工具创建和测试： 的Python 3.5 TensorFlow 1.4 配套 张量流 麻木 tqdm（可以通过删除utils.save_mnist_as_image（）中mnist_batch周围的tqdm包装器，并在CapsNetEMModel.learn_from_epoch（）中将trange（）更改为range（）来消除此要求。 matplotlib（仅在通过utils.save_mnist_as_image（）产生附加结果时才需要，而该模型可以不使用） pip install -r requirements.txt 或对于GPU TensorFlow： pip install -r requirements-gpu.txt 用法 要使用默认参数在MNIST上训练模型： python main.py