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