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)