ubuntu上使用draw_net.py绘制caffe ssd网络结构

1.先安装依赖库

安装graphviz：

   sudo apt-get install graphviz

安装pydot:

先安装pip

sudo apt-get install python-pip python-dev build-essential 

sudo pip install --upgrade pip 

sudo pip install --upgrade virtualenv

再安装pydot
sudo pip install pydot

python的pydot模块是Graphviz的dot语言的python接口，主要使用Graphviz绘图，当然要安装pydot

2. draw
python ./python/draw_net.py  ./models/VGGNet/VOC0712/SSD_300x300/train.prototxt   ssd_net.png --rankdir=LR
                                                                 模型                                                                     保存的图片      
备注：LR 从左到右
如果图片是bmp格式会出错：   return get_pydot_graph(caffe_net, rankdir, phase=phase).create(format=ext)
  File "/usr/local/lib/python2.7/dist-packages/pydot.py", line 1900, in create
    assert p.returncode == 0, p.returncode
AssertionError: 1




或者 python ./python/draw_net.py  ./examples/mnist/lenet_train_test.prototxt  test.png