Skip to content

Latest commit

 

History

History
30 lines (24 loc) · 773 Bytes

readme.md

File metadata and controls

30 lines (24 loc) · 773 Bytes

Capsule Net implementation in Gluon

This is a Gluon implementation of Capsule Net.

Dependency

  • mxnet-cu80==0.12.0
  • easydict==1.6
  • visdom==0.1.6.3

you can fastly install them by command

pip install -r requirements.txt

Usage

python train.py

Evaluation

The validation accuracy is about at 99.53% (mybe need more data augmentation).
The first three training epochs and validation graph (by visdom)

Reference

XifengGuo/CapsNet-Keras
Soonhwan-Kwon/capsnet.mxnet