GitHub - bourdakos1/capsule-networks: A Tensorflow implementation of Capsule Net...
source link: https://github.com/bourdakos1/capsule-networks
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
Capsule Networks
A Tensorflow implementation of Capsule Networks
For more information on Capsule Networks please read my Medium article.
[Update] Check out my new article with a deeper explanation on how capsules work.
This repo was largely based off of naturomics CapsNet-Tensorflow repo with the current differences being an added requirements.txt
and the MNIST dataset is included. Power users may want to refer to the naturomics repo for the latest versions.
Usage
Install the requirements and download dataset:
pip install -r requirements.txt
python download_data.py
Start training!
python main.py
Testing Accuracy
python main.py --is_training False
Visualization
Check out this Visualization Tool I built to play around with the DigitCaps vectors to see how it effects the recontructions:
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK