文件名称：MatrixCapsulesWithEMRouting:在TensorFlow中实现``具有EM路由的矩阵胶囊''
文件大小：60KB
文件格式：ZIP
更新时间：2024-06-03 20:19:32
Python
具有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
【文件预览】：
MatrixCapsulesWithEMRouting-master
----models()
--------capsule_layers.py(7KB)
--------capsnetem_model.py(13KB)
--------capsule_ops.py(28KB)
--------base_model.pyc(6KB)
--------utils.py(3KB)
--------__init__.pyc(963B)
--------__pycache__()
--------__init__.py(569B)
--------base_model.py(8KB)
--------tf_ops.py(30KB)
----main.py(8KB)
----requirements-gpu.txt(36B)
----results()
--------visualisations.py(484B)
----requirements.txt(32B)
----tests.py(23KB)
----README.md(6KB)