conviz icon indicating copy to clipboard operation
conviz copied to clipboard

An example of visualization of weights and output of convolutional layers

Conviz

Conviz is an example of visualizing weights and output of convolutional layers. In this project I show how to add convolution kernels and output of convolutional layers to a collection that fetched and visualized after training is completed. This approach can be easily generalized to arbitrary number of layers.

It is based on the following tutorials:

Data set

MNIST data set that is shiped with Tensorflow.

Output

Plotted convolution kernels and output of convolutional layers are saved to disk.

First convolutional layer

con_weights_0 con_output_0

Second convolutional layer

con_weights_1 con_output_1

Dependencies

  • Numpy
  • Matplotlib
  • Tensorflow