Capsule-master
所属分类:人工智能/神经网络/深度学习
开发工具:Python
文件大小:3KB
下载次数:18
上传日期:2019-01-08 18:04:15
上 传 者:
pafc882
说明: 深度学习之父新提出的动态路由胶囊网络的 keras实现
(capsnet in keras you can use this code to start your demo)
文件列表:
Capsule_Keras.py (3232, 2018-03-19)
capsule_test.py (3921, 2018-03-19)
# Capsule
动态路由算法来自:https://kexue.fm/archives/5112
该版本的动态路由略微不同于Hinton原版,在“单数字训练、双数字测试”的准确率上有95%左右。
其他:
1、相比之前实现的版本:https://github.com/XifengGuo/CapsNet-Keras ,我的版本是纯Keras实现的(原来是半Keras半tensorflow);
2、通过K.local_conv1d函数替代了K.map_fn提升了好几倍的速度,这是因为K.map_fn并不会自动并行,要并行的话需要想办法整合到一个矩阵运算;
3、其次我通过K.conv1d实现了共享参数版的;
4、代码要在Keras 2.1.0以上版本运行。
近期下载者:
相关文件:
收藏者: