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' 错误
生成网络结构图片