Skip to main content

A CNN model visualizer

Project description

Picasso

https://img.shields.io/pypi/v/picasso-viz.svg https://img.shields.io/travis/merantix/picasso.svg Documentation Status https://img.shields.io/codecov/c/github/merantix/picasso/master.svg

A CNN model visualizer

See the Medium post for an introduction to Picasso.

If you use Picasso in your research, please cite our paper:

@misc{picasso2017,
      Author = {Ryan Henderson and Rasmus Rothe},
      Title = {Picasso: A Neural Network Visualizer},
      Year = {2017},
      Eprint = {arXiv:1705.05627},
      Url = {https://arxiv.org/abs/1705.05627}
     }

Quickstart

Picasso uses Python 3.5+ so use a virtual environment if necessary (e.g. virtualenv env --python=python3) and activate it!

  1. Install with pip or from source.

    With pip:

    pip install picasso-viz

    From the repository:

    git clone git@github.com:merantix/picasso.git
    cd picasso
    pip install -e .

    Note: you’ll need the Tensorflow backend for Keras for these examples to work. Make sure your ~/.keras/keras.json file looks like:

    {
        "backend": "tensorflow",
        "image_dim_ordering": "tf",
        "floatx": "float32",
        "epsilon": 1e-07
    }
  2. Start the Flask server

    export FLASK_APP=picasso
    flask run

    Point your browser to 127.0.0.1:5000 and you should see the landing page! When you’re done, Ctrl+C in the terminal to kill your Flask server.

Building the docs

The documentation is much more extensive than this README, and includes instructions on getting the Keras VGG16 and Tensorflow NMIST models working, as well as guides on building your own visualizations and using custom models. This assumes you’ve cloned the repository. First install the required packages:

pip install -e .[docs]

Then build them:

cd docs/
make html

Then you can open _build/html/index.html in your browser of choice.

Notes

  1. Models generated on Keras using the Theano backend should in principle be supported. The only difference is the array ordering of convolutions. I haven’t tried this yet though, so an extra config parameter may be needed.

Credits

History

0.2.0 (2017-06-07)

  • Add RESTful API

  • Rework configuration loaders

  • Major refactor

0.1.2 (2017-06-07)

  • Fix Keras loading issues

  • Check tensorflow installation before installing

0.1.1 (2017-05-16)

  • First release on PyPI.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

picasso_viz-0.2.0.tar.gz (9.4 MB view hashes)

Uploaded Source

Built Distribution

picasso_viz-0.2.0-py2.py3-none-any.whl (8.9 MB view hashes)

Uploaded Python 2 Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page