资源分类
》 资源 》算法》CapsNet-CNTK

CapsNet-CNTK

2020-03-27 | |  50 |   0 |   0

CapsNet-CNTK

A CNTK implementation of CapsNet based on the paper titled 'Dynamic Routing Between Capsules' by Sara Sabour, Nicholas Frost and Geoffrey Hinton.

Requeriments

Architecture

CapsNetArch.png

Training

git clone https://github.com/southworkscom/CapsNet-CNTK.git
cd CapsNet-CNTK
python get_data.py
python main.py

Tensorboard

Visualize training progress with tensorboard:

tensorboard --logdir tensorboard

Then, Open your favorite browser and navigate to http://localhost:6006 (assuming you are running the training at localhost)

Results

Test Error

MethodRoutingReconstructionMNIST (%)Paper
Baseline------0.39
CapsNet3no0.550.35 (0.036)
CapsNet3yesWIP0.25 (0.005)

Reconstruction layer

reconstruction.png

Dimension perturbations

图片.png

TODO

  • Improve performance on CPU

  • Complete benchmarks

  • Capsule visualization

  • Improve the API

  • Try other datasets


上一篇： zeroshot-capsnet-pytorch

下一篇：lanes-capsnet

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...