Caffe中网络的绘制

1、安装graphviz:

 sudo apt-get install graphviz  #依赖库

2、安装pydot

sudo pip install pydot

3、使用命令行进行网络的绘制

#第一个参数是Caffe文件中Python文件夹下的draw_net.py文件
#第一个参数：网络模型的prototxt文件
#第二个参数：保存图像的路径及名称
#第三个参数：--rankdir=x,y 有四种选项，分别是LR,RL,TB,BT,分别是从左到右，从右到左，从上到下，从下到上。默认为LR

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

这样绘制好的网络模型就存放在设定的路径下了！

问题：
1 ImportError: No model named skimage.io

解决办法：sudo pip install scikit-image