group_equivariant_capsules_pytorch icon indicating copy to clipboard operation
group_equivariant_capsules_pytorch copied to clipboard

Published 20 hours ago • verified publisher icon janericlenssen
→

Metadata

Pytorch implementation of Group Equivariant Capsule Networks

Group Equivariant Capsule Networks in PyTorch

Proof of concept Pytorch implementation of Group Equivariant Capsule Networks (NIPS 2018).

Required:

  1. PyTorch 0.4.1
  2. PyTorch Geometric
  3. pytorch_unique

Install:

  1. Clone Repo
  2. python setup.py install

In repo dir: python examples/mnist.py to run the MNIST Experiment.

Implementations of capsules in pooling_capsule_layer.py

Implementations of sparse group convolution in group_conv_layer.py

← Metadata

29
Stars
8
Forks
Watchers

Owner

verified publisher icon janericlenssen

Metadata

Pytorch implementation of Group Equivariant Capsule Networks

Back