Skip to content

michaelklachko/CapsNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 

Repository files navigation

CapsNet

A TensorFlow implementation of Hinton's "Dynamic Routing Between Capsules" paper, using CIFAR-10 dataset

To run, first download CIFAR-10 dataset from https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz extract cifar-10-batches-py folder and put it in the same dir as capsnet_cifar.py (or change dataset path in the code).

On a single 1080Ti, training with default parameters takes 15 min per epoch.

TODO: implement "none of the above" class.

About

Implementation of Hinton's "Dynamic Routing Between Capsules" paper

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages