search

naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules
Apache License 2.0
3.8k stars 1.17k forks source link

Adding layers #54

Open amobiny opened 6 years ago

amobiny commented 6 years ago

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?

naturomics commented 6 years ago

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(...)