文件名称:capsule-networks:胶囊网络的Tensorflow(v1.4)实现(胶囊之间的动态路由,https
文件大小:11.19MB
文件格式:ZIP
更新时间:2024-06-07 00:56:26
Python
胶囊网络的Tensorflow实现 要求 tensorflow> = 1.4(也许1.3也应该工作,尽管我没有测试它) 麻木 枕头 科学的 胶囊网络 我试图实现这个想法 文件描述 config.py包含所有需要的超级参数。 utils.py包含有关加载和保存的功能。 model.py具有capsNet和整个模型实现的所有构建块。 train.py用于培训。 eval.py用于评估。 用法 训练 $ python train.py 评估 $ python eval.py 训练 结果 分类 时代 10 测试权限 99.278 重建
【文件预览】:
capsule-networks-master
----figure()
--------test_005.png(22KB)
--------capsNet.png(128KB)
--------margin_loss.png(11KB)
--------total_loss.png(11KB)
--------test_000.png(21KB)
--------reconstruction_loss.png(14KB)
--------test_010.png(22KB)
----.gitignore(1KB)
----capsNet()
--------eval.py(2KB)
--------config.py(146B)
--------train.py(2KB)
--------utils.py(3KB)
--------model.py(12KB)
--------__init__.py(0B)
--------data()
----README.md(1KB)
----LICENSE(11KB)
----__init__.py(0B)
----setup.py(319B)