Giters

naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Adding layers

amobiny opened this issue · comments

commented

Thanks for the nice code. I had a question regarding the capsnet. Would it be possible to add layers (like a conv-caps layer after the first primary layer, or a fully connected caps layer with 20 capsules befor the digit caps layer?)
I tried it myself and am getting terrible results! I can't understand why is it happening. Do you have any idea?

commented

Yes, I created a new project to meet your situation. Please see my CapsLayer repo. You can import capslayer and then call it in your code: capslayer.layers.conv2d(...) or capslayer.layers.fully_connected(...)