/CapsNets

PyTorch Implementation of Capsule Networks

Primary LanguageJupyter Notebook

Dynamic Routing between capsules

PyTorch implementation of Dynamic Routing Between Capsules

TODO

  • Implement Layers

    • Conv Layer
    • Primary Capsules Layer
    • Digi Capsules Layer
    • [] Decoding Layer

  • Losses

    • [] Margin Loss
    • [] Reconstruction Loss

  • Datasets

    • [] MNIST
    • [] CIFAR 10

[] Add Performance Metrics [] Add visualizations from layers [] Write A report