Skip to content

Godricly/CapsNet

Repository files navigation

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages