search

pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
15.92k stars 6.91k forks source link

CapsNet Model #375

Open ahkarami opened 6 years ago

ahkarami commented 6 years ago

Recently Geoffrey Hinton and his team introduced a new interesting deep neural network model, called CapsNet. I think the implementation of this Net on PyTorch and train it on ImageNet Data set, and after that add its pre-trained model to TorchVision Models would be great for the PyTorch community.
Thanks to the PyTorch team

Lyken17 commented 6 years ago

As far as I know, Hinton's group doesn't show experiments on large-scale dataset -- they only tested MNIST.