Skip to content

Latest commit

 

History

History
15 lines (9 loc) · 561 Bytes

README.md

File metadata and controls

15 lines (9 loc) · 561 Bytes

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.