python绘制caffe中网络模型

caffe-master/python/draw_net.py 实现绘制caffe中定义的网络模型功能，将.prototxt文件可视化。

需要先安装pydot和protobuf工具

通过Anaconda安装pydot和protobuf工具：

sduo chmod 777 -R ~/anaconda2
conda install protobuf
conda install pydot

如果还缺少其他工具，也可以通过这种方式安装。

绘制Lenet网络模型

MNIST手写体网络模型Lenet：

cd ~/caffe-master
./python/draw_net.py ./examples/mnist/lenet_train_test.prototxt ./python/lenet.png

cifar：

cd ~/caffe-master
./python/draw_net.py ./examples/cifar10/cifar10_quick_train_test.prototxt ./python/cifar.png

更改网络模型方向显示

draw_net.py带三个参数:

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

• 第二个参数：保存的图片路径及名字

• 第二个参数：--rankdir=x , x 有四种选项，分别是LR, RL, TB, BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为LR。

更改绘制方向为从下到上：

Lenet：

cifar：