yechengxi / LightCapsNet

A Matlab implementation of the capsule networks (or capsnet).

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

LightCapsNet(testing stage)

A Matlab implementation of CapsNet in the paper:
Sara Sabour, Nicholas Frosst, Geoffrey E Hinton. Dynamic Routing Between Capsules. NIPS 2017

The network is implemented in the hard way. The back propagation is first derived by hand and then implemented.

At this time the an MLP capsule network and CNN capsult network is implemented in the tests. The 'dynamic routing' is still in development. One simple example is provided and it is likely to be a new hardware killer.

Collaborations and contributions are very welcome.

This project is based on LightNet.

About

A Matlab implementation of the capsule networks (or capsnet).

License:GNU General Public License v3.0


Languages

Language:MATLAB 100.0%