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