xiaochus / VisualizationCNN

Visualization CNN model by Keras.

MIT License

73 stars 32 forks source link

cnn-keras visualization

readme

VisualizationCNN

Visualization CNN model by Keras.

Requirement

Python 3.6
Keras 2.2.2
Tensorflow-gpu 1.8.0
OpenCV 3.4

Support model

VGG16
VGG19
ResNet50
InceptionV3
Xception
MobileNet
DenseNet121

Visualization (VGG16 example)

Visualizing intermediate conv outputs conv_block1_conv1 conv_block2_conv2 conv_block3_conv3

Visualizing conv filters filter_block1_conv1 filter_block2_conv2

Visualizing heatmaps of class activation in an image heatmap