Godricly / CapsNet

CapsNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A MxNet implementation of Capsule

How to use

Train

Model training

python train.py

Test

Show reconstruction result and interpolation result

python test.py

ToDo:

  • revised ReadMe ???
  • more fancy plots???

Rec Result

input rec output
alt text alt text

Interpolation

Each row shows interpolation of one digit alone different channel.
alt text

Reference

Capsule_MxNet Some codes are cropped from this repo, However I'm concerned with its implementation

Capsule_Keras Very nice Keras implemtation which clarifies some concepts. I'm still confused with the stop gradient parts.

MxNet pull #8787 A Capsule implementation which uses old symbol system.

About

CapsNet

License:MIT License


Languages

Language:Python 100.0%