Skip to content

Ugenteraan/CapsNet-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 

Repository files navigation

CapsNet in PyTorch 1.4

A Capsule Network with Dynamic routing based on https://arxiv.org/abs/1710.09829 for MNIST classification.

Figure below shows the result of the reconstruction network. From the figure, it can be clearly seen that the Capsule Network was able to learn the features of the digits such as orientation, thickness and skewness.

About

Capsule Network with Dynamic Routing using PyTorch version 1.4

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published