可视化神经网络特征_超方便

Github链接:GitHub-JonnesLin/Evison: We provide an easy way for visualizing——https://github.com/JonnesLin/Evison

首先需要安装Evison：

pip install Evison

示例：

from Evison import Display, show_network
from net import CNN_3D

# 生成我们需要可视化的网络(可以使用自己设计的网络)
network = CNN_3D(16)

# 使用show_network这个辅助函数来看看有什么网络层(layers)
show_network(network)

# block_3D
# block_3D.0
# block_3D.1
# block_3D.2
# block_3D.3
# block_3D.4
# block_3D.5
# classifier
# classifier.0

# 构建visualization的对象 以及 制定可视化的网络层
visualized_layer = 'block_3D.5'
display = Display(network, visualized_layer, img_size=(224, 224))  # img_size的参数指的是输入图片的大小

# 加载我们想要可视化的图片
from PIL import Image
image = Image.open('000000000285.jpg').resize((224, 224))

# 将想要可视化的图片送入display中，然后进行保存
display.save(image)