【文件属性】:
文件名称:capsule-networks:胶囊网络的Tensorflow(v1.4)实现(胶囊之间的动态路由,https
文件大小:11.19MB
文件格式:ZIP
更新时间:2021-05-18 07:09:46
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)