胶囊网络CapNet从PrimaryCaps到DigitCaps是怎么实现的?
胶囊网络CapNet从PrimaryCaps到DigitCaps是怎么实现的?
先看一下代码,有注释,八成能懂.
这里假设batch_size=100, 下文的维度中的100都是指的batch_size.
首先PrimaryCaps中这里是通过8个普通卷积实现的.
从PrimaryCaps这里的特征图在上图中很清楚shape为[100, 32, 6, 6, 8],但是你可以把这个8看做是一个向量, 对于特征图中的一个点就可以用一个长度为8的向量表示,一共有多少这样的点呢? 3266个.
则Primar
原创
2020-07-28 18:09:10 ·
1413 阅读 ·
1 评论