• Stars
    star
    102
  • Rank 335,584 (Top 7 %)
  • Language
    Python
  • Created about 8 years ago
  • Updated about 8 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

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