Dynamic Routing Between Capsules 复现

Capsules Network


​ 上一次看这篇论文:Dynamic Routing Between Capsules 的时候,Pytorch技术还很菜,不是特别熟练,被矩阵计算的维数问题搞傻了,写得很痛苦,loss没写完就放弃了。这次在重新读完论文之后,重新试着复现了一波,最后差点败在一个softmax的执行维度上(可以说很奇异了)。第一版caps net网络结构非常简单,对应的evaluation / experiments也很简单。实现见[Github🔗Enigmatisms/CapsNet]

G_1101 G_1151
根据digital capsule重建(MNIST 7 epochs) 根据digital capsule重建(MNIST 7 epochs)

​ 等待填坑