Need help with capsule-networks?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

JunYeopLee
204 Stars 36 Forks Apache License 2.0 30 Commits 0 Opened issues

Description

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

Services available

!
?

Need anything else?

Contributors list

# 322,619
Python
28 commits

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

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.