/CapsNet-Visualization

🎆 A visualization of the CapsNet layers to better understand how it works

Primary LanguagePythonMIT LicenseMIT

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]