← all repositories

gram-ai/capsule-networks

A PyTorch implementation of the CapsNet architecture for image classification using dynamic routing between capsules.

1.8k stars Python ML Frameworks
capsule-networks
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository implements the CapsNet architecture from the 2017 paper by Sabour, Frosst, and Hinton. It trains a multi-layer capsule network on MNIST for digit recognition, using an iterative routing-by-agreement mechanism where lower-level capsules send outputs to higher-level capsules based on vector agreement. The implementation supports CUDA acceleration and uses TorchNet for training orchestration.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.