Skip to content

GarrickLin/Capsnet.Gluon

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Capsule Net implementation in Gluon

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages