caffe学习：绘制网络模型

在确保caffe的Python接口编译正确的情况下

1  安装GraphViz

sudo apt-get install GraphViz

2 安装pydot

sudo pip install pydot

安装好后就可以调用caffe中Python文件夹下的draw_net.py脚本画图了

 比如绘制caffe自带的LeNet网络模型

在caffe文件夹下输入：

sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt  lenet.png --rankdir=LR

说明：draw_net.py执行的时候带三个三数

第一个参数：网络模型文件

第二个参数：保存的图片的名字

第三个参数：--rankdir=LR或RL/TB/BT 用来表示网络的方向