Skip to content

Capsule Network code using TensorFlow Estimator API

Notifications You must be signed in to change notification settings

piyush2896/CapsuleNetworks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published