Skip to content

A Tensorflow(v1.4) implementation of Capsule Networks (Dynamic Routing Between Capsules , https://arxiv.org/abs/1710.09829)

License

Notifications You must be signed in to change notification settings

JunYeopLee/capsule-networks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

30 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

A Tensorflow implementation of Capsule networks

Requirements

  • tensorflow >= 1.4 (Probably 1.3 should work, too, though I didn't test it)
  • numpy
  • pillow
  • scipy

Capsule network

I tried to implement the idea in Dynamic Routing Between Capsules

File description

  • config.py includes all hyper parameters that are needed.
  • utils.py contains functions regarding loading and saving.
  • model.py has all building blocks for capsNet and whole model implementation.
  • train.py is for training.
  • eval.py is for evaluation.

Usage

Training

$ python train.py

Evaluation

$ python eval.py

Training

Result

Classification

Epoch 10
Test Acc. 99.278

Reconstruction

About

A Tensorflow(v1.4) implementation of Capsule Networks (Dynamic Routing Between Capsules , https://arxiv.org/abs/1710.09829)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages