censomin / capsnet-fashion-mnist

Capsule Network on Fashion MNIST dataset

CapsNet-Fashion-MNIST

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.