piyush2896 / CapsuleNetworks

Capsule Network code using TensorFlow Estimator API

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Capsule Networks

Capsule Networks are networks that are considered to perform "inverse graphics". This means that using the image in the view they give us the embeddings which can be used to reconstruct the image perfectly well.

In this repository, I created a capsule network inspired by this paper. The network is built entirely with tensorflow's estimator API.

Architecture

The architecture of the network is as shown below.

capsule nets

Reconstruction Results

results

Graph As Shown on TensorBoard

tensorboard graph

Dependencies

  1. TensorFlow
  2. Matplotlib
  3. Numpy

About

Capsule Network code using TensorFlow Estimator API


Languages

Language:Jupyter Notebook 79.1%Language:Python 20.9%