capsule-networks

by JunYeopLee

JunYeopLee / capsule-networks

A Tensorflow(v1.4) implementation of Capsule Networks (Dynamic Routing Between Capsules , https://ar...

202 Stars 36 Forks Last release: Not found Apache License 2.0 30 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

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.