CapsNet-pytorch

Capsule network library

An implementation of a deep learning model for image classification using capsule networks with dynamic routing.

PyTorch implementation of NIPS 2017 paper Dynamic Routing Between Capsules

GitHub

489 stars
11 watching
73 forks
Language: Python
last commit: over 3 years ago
capsnetcapsulesdeep-learningdynamic-routing-between-capsulesmachine-learningmnistpytorch

Related projects:

Repository Description Stars
timomernick/pytorch-capsule An implementation of Hinton's Dynamic Routing Between Capsules in Pytorch 632
benedekrozemberczki/capsgnn A PyTorch implementation of a graph neural network architecture 1,246
joel-huang/zeroshot-capsnet-pytorch An implementation of Zero-shot User Intent Detection via Capsule Neural Networks using PyTorch. 16
yechengxi/lightcapsnet An implementation of Capsule Networks in Matlab for image classification and other computer vision tasks 48
higgsfield/capsule-network-tutorial An easy-to-follow PyTorch tutorial on building and understanding Capsule Networks using Jupyter Notebook. 768
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112
chenxi116/pnasnet.pytorch PyTorch implementation of PNASNet-5 architecture 317
kimhc6028/relational-networks A PyTorch implementation of a neural network module for relational reasoning in computer vision tasks 812
tobiascz/mnist_pytorch_python_and_capi An example project demonstrating how to train and deploy a neural network in Python and C++ using PyTorch 1.0 96
fxia22/pointnet.pytorch This is an implementation of the PointNet algorithm in PyTorch for 3D point cloud classification and segmentation tasks. 2,149
dyhan0920/pyramidnet-pytorch An implementation of a deep neural network architecture for image classification tasks 273
kazuto1011/pspnet-pytorch Re-implementation of a deep learning model for semantic segmentation using PyTorch. 52
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
wandering007/nasnet-pytorch An implementation of NASNet in PyTorch using ported weights from TensorFlow 75
oyam/pytorch-dpns PyTorch implementation of a deep learning model for image segmentation 90