Skip to content

bourdakos1/CapsNet-Visualization

Repository files navigation

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]