Caffe---自带工具进行网络结构(xxx.prototxt)可视化

Caffe---自带绘图工具(draw_net.py)绘制网络结构图(xxx.prototxt)

目录：

一，安装依赖库。

二，draw_net.py使用说明。

 

正文：

一，安装依赖库。

在绘制之前，需要先安装相关的依赖库
１、安装ＧraphViz
# sudo apt-get install GraphViz
注意，这里用的是apt-get来安装，而不是pip.

2 、安装pydot
# sudo pip install pydot
用的是pip来安装，而不是apt-get
安装好了，就可以调用脚本来绘制图片了

二，draw_net.py使用说明。
draw_net.py执行的时候带三个参数
第一个参数：网络模型的prototxt文件
第二个参数：保存的图片路径及名字
第二个参数：--rankdir=x , x 有四种选项，分别是LR, RL, TB, BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为ＬＲ。

例：绘制Lenet模型
sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=LR

#cd caffe
#python python/draw_net.py examples/mnist/lenet_train_test.prototxt my_show_NetStructure/mnist_lenet_train_test.png --rankdir=LR

#python python/draw_net.py examples/mnist/mnist_autoencoder.prototxt my_show_NetStructure/mnist_autoencoder.png --rankdir=LR

#python python/draw_net.py examples/cifar10/cifar10_quick_train_test.prototxt my_show_NetStructure/cifar10_quick_train_test.png --rankdir=LR

 

最后补充，在Caffe中可视化prototxt格式网络结构的方法：目前除了可以使用Caffe提供的【1】draw_net.py，还可以使用【2】Netscope在线可视化。

Netscope是个支持prototxt格式描述的神经网络结构的在线可视工具，即:http://ethereon.github.io/netscope/quickstart.html 。

使用方法：首先打开这个地址:http://ethereon.github.io/netscope/#/editor，然后把你的描述神经网络结构的prototxt文件复制到该编辑框里，最后按shift-enter，这样就可以直接以图形方式显示网络的结构 。

使用Netscope在线可视化，本人亲自玩了几次，发现网址不好用了。