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 5 years ago
capsnetcapsulesdeep-learningdynamic-routing-between-capsulesmachine-learningmnistpytorch

Related projects:

RepositoryDescriptionStars
timomernick/pytorch-capsuleAn implementation of Hinton's Dynamic Routing Between Capsules in Pytorch633
benedekrozemberczki/capsgnnA PyTorch implementation of a graph neural network architecture1,246
joel-huang/zeroshot-capsnet-pytorchAn implementation of Zero-shot User Intent Detection via Capsule Neural Networks using PyTorch.16
yechengxi/lightcapsnetAn implementation of Capsule Networks in Matlab for image classification and other computer vision tasks48
higgsfield/capsule-network-tutorialAn easy-to-follow PyTorch tutorial on building and understanding Capsule Networks using Jupyter Notebook.770
4uiiurz1/pytorch-res2netImplementations of deep learning architectures using PyTorch for image classification tasks on various datasets.112
chenxi116/pnasnet.pytorchPyTorch implementation of PNASNet-5 architecture317
kimhc6028/relational-networksA PyTorch implementation of a neural network module for relational reasoning in computer vision tasks809
tobiascz/mnist_pytorch_python_and_capiAn example project demonstrating how to train and deploy a neural network in Python and C++ using PyTorch 1.096
fxia22/pointnet.pytorchThis is an implementation of the PointNet algorithm in PyTorch for 3D point cloud classification and segmentation tasks.2,175
dyhan0920/pyramidnet-pytorchAn implementation of a deep neural network architecture for image classification tasks273
kazuto1011/pspnet-pytorchRe-implementation of a deep learning model for semantic segmentation using PyTorch.52
zhanghang1989/pytorch-encodingA Python framework for building deep learning models with optimized encoding layers and batch normalization.2,044
wandering007/nasnet-pytorchAn implementation of NASNet in PyTorch using ported weights from TensorFlow75
oyam/pytorch-dpnsPyTorch implementation of a deep learning model for image segmentation90