# CapsNet-Visualization

![](readme/4.gif)

For more information on capsule networks check out my Medium articles [here](https://hackernoon.com/capsule-networks-are-shaking-up-ai-heres-how-to-use-them-c233a0971952) and [here](https://medium.freecodecamp.org/understanding-capsule-networks-ais-alluring-new-architecture-bdb228173ddc).

## Setup

Use pip to install the required python packages:

```bash
pip install -r requirements.txt
```

## Running the Tool

Start the Flask application by running:

```bash
python run_visualization.py
```

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

## Testing your own Images

Add your images to the `test_images` directory.

Run:
```bash
python render.py [filename]
```