Matrix capsules with EM routing 论文笔记

1、网络架构

注：A=B=C=D=32，E=类别

层	参数	输出
MNIST	图片	[28,28,1]
ReLU_Conv1	kernel_size=[5,5] stride=2, padding=same,output=32	[14,14,32]
PrimaryCaps	kernel_size=[1,1] stride=1, padding=valid,output=32	pose[14,14,32,4,4],activation[14,14,32,1]
ConvCaps1	kernel_size=[3,3] stride=2, padding=valid,output=32	pose[6,6,32,4,4],activation[6,6,32,1]
ConvCaps2	kernel_size=[3,3] stride=2, padding=valid,output=32	pose[4,4,32,4,4],activation[4,4,32,1]
Class Capsules	kernel_size=[1,1] stride=2, padding=valid,output=32

2、计算胶囊的pose和activation

在EM路由中，用高斯模型对L+1胶囊的姿态矩阵进行建模。姿态矩阵[4，4]，16个元素。用具有16个μ和16个σ的高斯模型对姿态矩阵建模，每个μ表示姿态矩阵的一个元素。

 

 

 

持续更新.....