πŸŽ† A visualization of the CapsNet layers to better understand how it works
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
client build Jan 31, 2018
example_weights_json init Jan 19, 2018
numpy_weights reconstructions Jan 26, 2018
readme update Feb 13, 2018
reconstructions Revert "Merge pull request #1 from jiayiliu/master" Mar 13, 2018
test_images update Feb 13, 2018
visualizations update Feb 13, 2018
.gitignore Revert "Merge pull request #1 from jiayiliu/master" Mar 13, 2018
LICENSE Initial commit Jan 19, 2018
README.md Update README.md Feb 13, 2018
extract_weights.py reconstructions Jan 26, 2018
render.py update Feb 13, 2018
requirements.txt Update requirements.txt Mar 13, 2018
run_visualization.py lots of updates Jan 31, 2018

README.md

CapsNet-Visualization

For more information on capsule networks check out my Medium articles here and here.

Setup

Use pip to install the required python packages:

pip install -r requirements.txt

Running the Tool

Start the Flask application by running:

python run_visualization.py

Point your browser to: http://localhost:5000

Testing your own Images

Add your images to the test_images directory.

Run:

python render.py [filename]