文件名称:CapsNet-Keras:NIPS2017论文“胶囊之间的动态路由”中CapsNet的Keras实现。 现在测试误差= 0.34%
文件大小:590KB
文件格式:ZIP
更新时间:2024-03-11 20:29:47
keras capsule dynamic-routing capsnet capsule-network
CapsNet-Keras 本文中的 (分支支持TensorFlow 2)实现: 当前的average test error = 0.34% , best test error = 0.30% 。 与论文的区别: 我们使用decay factor = 0.9且step = 1 epoch的学习率衰减, 而论文没有给出详细的参数(或者他们没有使用它?)。 我们仅在训练了50 epochs后才报告测试错误。 在本文中,我想他们根据图A.1训练了1250 epochs ? 听起来很疯狂,也许我误会了。 我们使用MSE(均方误差)作为重建损失,损失系数为lam_recon=0.0005*784=0.392 。 与本文中使用SSE(平方和误差)和lam_recon=0.0005等效。 警告 请使用Keras == 2.0.7和TensorFlow == 1.2后端,否则K.batch
【文件预览】:
CapsNet-Keras-master
----utils.py(2KB)
----result()
--------manipulate-6.png(47KB)
--------real_and_recon.png(22KB)
--------manipulate-8.png(45KB)
--------manipulate-2.png(59KB)
--------manipulate-9.png(40KB)
--------manipulate-1.png(20KB)
--------manipulate-4.png(44KB)
--------manipulate-5.png(57KB)
--------model.png(77KB)
--------manipulate-0.png(49KB)
--------manipulate-7.png(34KB)
--------log.png(43KB)
--------manipulate-3.png(52KB)
----LICENSE(1KB)
----capsulenet-multi-gpu.py(6KB)
----capsulelayers.py(10KB)
----.gitignore(50B)
----README.md(6KB)
----capsulenet.py(11KB)