DeppMeng / conviz

An example of visualization of weights and output of convolutional layers

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

An example of visualization of weights and output of convolutional layers


Languages

Language:Python 100.0%