1、安装graphViz
打开网址,下载2.38稳定版本的graphViz
点击下载graphviz-2.38.msi,下载完成后,双击msi文件进行安装,安装完成之后,配置环境变量,我的电脑默认安装在:C:\Program Files (x86)\Graphviz2.38该目录下,将该路径C:\Program Files (x86)\Graphviz2.38\bin配置环境变量。
验证: 进入windows命令行界面,输入dot -version,然后按回车,如果显示graphviz的相关版本信息,则安装配置成功。
2 、安装pydot和pydotplus
pip install pydot
pip install pydotplus
3、代码实现绘制结构图
python D:\Caffe\caffe-windows\python\draw_net.py D:\LSLS\caffe\alexnet_lmdb\alexnet_train.prototxt D:\LSLS\caffe\alexnet_train.png --rankdir=BT
直接在pycharm里运行draw_net.py,配置好参数,运行能出结构图。但是在cmd终端执行上述命令就会报错,最终发现是在终端import caffe出了问题,但是在pycharm里面import caffe没有报错
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第二个参数:–rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。