/conviz

An example of visualization of weights and output of convolutional layers

Primary LanguagePython

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