Skip to main content
Help us improve PyPI by participating in user testing. All experience levels needed!

A CNN model visualizer

Project description

Picasso

Documentation Status

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


Release history Release notifications

This version
History Node

0.2.0

History Node

0.1.2

History Node

0.1.1

Download files

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

Filename, size & hash SHA256 hash help File type Python version Upload date
picasso_viz-0.2.0-py2.py3-none-any.whl (8.9 MB) Copy SHA256 hash SHA256 Wheel py2.py3 Jul 20, 2017
picasso_viz-0.2.0.tar.gz (9.4 MB) Copy SHA256 hash SHA256 Source None Jul 20, 2017

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging CloudAMQP CloudAMQP RabbitMQ AWS AWS Cloud computing Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page