在线可视化caffe模型结构

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

假设Caffe的目录是$(CAFFE_ROOT)

1.编译caffe的python接口

$ make pycaffe

2.装各种依赖

$ pip install pydot
$ sudo apt-get install graphviz

3.可视化模型

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

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

绘制googlenet模型

$ cd $(CAFFE_ROOT)/python/
$ ./draw_net.py ../models/bvlc_googlenet/deploy.prototxt bvlc_googlenet.caffemodel.jpg

路径自己改变，./draw_net.py后面接prototxt路径和可视化图像的保存路径。

4.最终的图像

绘制其他模型

#绘制Lenet模型
cd (CAFFE_ROOT)
python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT  #绘制cifar10模型
cd $(CAFFE_ROOT)
python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT

本文转自：http://blog.csdn.net/ws_20100/article/details/50443566