Skip to content

XifengGuo/CapsNet-Fashion-MNIST

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CapsNet-Fashion-MNIST

License

A Keras implementation of CapsNet in the paper:
Sara Sabour, Nicholas Frosst, Geoffrey E Hinton. Dynamic Routing Between Capsules. NIPS 2017

This code is adopted from CapsNet-Keras to test the performance of CapsNet on Fashion-MNIST

Contacts
Xifeng Guo
E-mail [email protected] or WeChat wenlong-guo.

Usage

Step 1. Install Keras 2.0.9 with TensorFlow backend.

pip install tensorflow-gpu
pip install keras==2.0.9

Step 2. Clone this repository to local.

git clone https://github.com/XifengGuo/CapsNet-Fashion-MNIST.git
cd CapsNet-Fashion-MNIST

Step 3. Train a CapsNet on Fashion-MNIST

Training with default settings:

$ python capsulenet.py

Data preprocessing:

  • scale pixel values to [0,1];
  • shift 2 pixels and horizontal flipping augmentation.

Results

Accuracy

Test Accuracy: 93.62%

Losses and accuracies:

Training Speed

About 120s / epoch on a single GTX 1070 GPU.

Reconstruction result

Top 5 rows are real images from MNIST and Bottom are corresponding reconstructed images.

Releases

No releases published

Packages

No packages published

Languages