参考博文: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' 错误

生成网络结构图片

caffe绘制网络结构图

分类:

技术点:

相关文章: