Skip to content

Riretta/CapsNet_CIFAR10

Repository files navigation

CapsNet_CIFAR10

Configuration of CapsNet for CIFAR10 for pytorch implementation.

file = Capsule Network:

Original code from https://github.com/higgsfield/Capsule-Network-Tutorial The network is initialized to work with CIFAR10 dataset and reach a test accuracy of 71%.

Still working on recostruction of the image.

This implementation of Capsules with CIFAR10 is missing the ensembled models required for 89% accuracy.

file= Capsule Network with ensemble

The code is expanded with n ensembled models to improve the performance with the cifar10 images.