胶囊网络的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
重建
2022-05-22 21:08:57
11.19MB
Python
1