caffe绘制网络结构图

参考博文：caffe入门学习（5）：绘制网络结构图

运行脚本文件draw_caffe.bat

python C:/caffe/python/draw_net.py  C:/caffe/examples/mnist/lenet_train_test.prototxt D:/lenetmap.png --rankdir=LR
pause

参数说明： 
第一个参数：网络定义文件 
第二个参数：生成图片的保存名称和路径 
第三个参数：表示网络绘制方向，有四种，LR 从左到右,RL 从右到左, TB 从上到下 ,BT 从下到上。

双击后运行脚本，结果提示少库。

ImportError：No module named pydot

用pip进行安装

pip install pydotplus

注意：安装pydot可能会有AttributeError: 'module' object has no attribute 'graphviz' 错误

生成网络结构图片