A Python2/PyTorch Implementation of the Geoffery Hinton's Capsule Network Architecture with Extensions for Multiple Datasets, Losses, and Decoders (MNIST, FashionMNIST, CIFAR10, CIFAR100)
- Katie Link
- Anant Bhargava
- Jin Young (Daniel) Sohn
Current and Future Endeavors
- MNIST results
- FASHION MNIST results
- CIFAR10 results
- CIFAR100 results
- Reconstruct images
- ROC Curve Visualization
- Implement BCE and Reconstruction Losses
- Implement Tensorboard visualization