Skip to main content

A high-level API for ConvNet visualizations in Keras

Project description

keravis

keravis is a high-level API for ConvNet visualizations in Keras. As of v1.0, it supports visualizations of

  1. Convolutional layer activations
  2. 2-dimensional feature space representations
  3. Saliency maps (vanilla backprop, guided backprop, and occlusion)
  4. Generated inputs that result in maximal class scores
  5. Patches in a set of images that maximally activate an intermediate neuron

with support for nested pretrained models.

This is a hobby project that was inspired by lecture 14 of Stanford's CS231n: Convolutional Neural Networks for Visual Recognition http://cs231n.stanford.edu/. It is not yet optimized for serious use (see keras-vis instead).

Installation

You can install keravis using pip

pip install keravis

Usage

Read the documentation

Sample Visualizations

Below are sample visualizations from a small convolutional network trained on MNIST

from keravis import feature_space
feature_space(model,X=x_test[:5000],y=y_test[:5000],kind='tsne')

MNIST_TSNE

from keravis import saliency_backprop
saliency_backprop(model,test_img,class_idx=7)

saliency_1

from keravis import saliency_guided_backprop
saliency_guided_backprop(model,test_img,class_idx=7)

saliency

from keravis import maximal_class_score_input
maximal_class_score_input(model,class_idx=5,dim=(28,28,1))

gradient_ascent_5

from keravis import maximally_activating_patches
maximally_activating_patches(model,'conv2d_1',X=x_test)

MNIST_CONV_FEATURES

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

keravis-1.0.2.tar.gz (14.8 kB view hashes)

Uploaded Source

Built Distribution

keravis-1.0.2-py3-none-any.whl (14.6 kB view hashes)

Uploaded 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