
Capsule Networks: 胶囊网络的TensorFlow (v1.4) 实现(含胶囊间动态路由,链接:https)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目提供了一个基于TensorFlow v1.4实现的胶囊网络框架,包含胶囊间的动态路由算法。了解更多详情和源码,请访问上述链接。
胶囊网络的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%。
全部评论 (0)
还没有任何评论哟~


