Caffe: Python 绘制网络模型

一、caffe/python/draw_net.py, 用来绘制网络模型
1、Anaconda 
sudo apt-get install graphviz
pip install pydot
conda install pydot-ng
conda install graphviz


2、Python
sudo apt-get install graphviz
pip install pydot


注意顺序，而且关注下pip 的版本和安装位置（系统自带/还是自己安装的Python/anaconda）
二、draw_net.py
第一个参数：网络模型的prototxt文件
第二个参数：保存的图片路径及名字
第二个参数：--rankdir=x , x 有四种选项，分别是LR, RL, TB, BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为ＬＲ。


（sudo）python python/draw_net.py examples/mnist/lenet_train_test.prototxt python/netImage/lenet.png --rankdir=BT