caffe net 可视化工具,,层特征可视化

时间:2022-11-23 13:55:14

1.只用网络在线结构绘制可视化网络模型

http://ethereon.github.io/netscope/#/editor

将对应的网络输入到里面,然后按shift+enter即可查看对应的网络结构

2,可以安装windows, linux, mac等平台,并且支持多种模型的可视化,包括caffe,tensorflow, ONNX等等

https://github.com/lutzroeder/netron

https://lutzroeder.github.io/netron/

2. 使用caffe自带的绘制工具

在caffe目录下的 python/draw_net.py这个文件,就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。

  在绘制之前,需要先安装两个库:GraphViz和pydot。 Graphviz是一个开源的可视化软件,能够以抽象的图和网络表示结构信息,广泛使用于网络,生物信息学,软件工程等领域。

  先安装graphviz否则会出现类似:"dot" not found in path 的问题

  安装graphviz不要用pip install安装,否则还是会找不到可执行程序

  安装:sudo apt-get insall graphviz

  然后安装pydot:sudo pip install pydot

draw_net.py执行的时候带三个参数:

  第一个参数:网络模型的prototxt文件 
  第二个参数:保存的图片路径及名字 
  第三个参数:–rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR

  # sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=TB

层的特征可视化

http://blog.csdn.net/qq_30401249/article/details/51469184