01 安装draw_net.py需要的必备组件
caffe提供话网络图的python工具caffe\python\draw_net.py
draw_net.py依赖 GraphViz pydot 两个Python组件。并且需要系统有GraphViz运行程序。
在命令行安装draw_net.py的python依赖组件。(我的系统使用的python3.5.3,如果使用python2.7.XX的使用pip2 install GraphViz Pydot)。
pip3 install GraphViz pydot
为操作系统安装GraphViz工具。
不安装 GraphViz 运行程序,运行draw_net.py 工具绘制网络图时,会提示如下错误:
Exception: “dot.exe” not found in path.
GraphViz下载网址:
http://www.graphviz.org/Download_windows.php
http://www.graphviz.org/pub/graphviz/stable/windows/graphviz-2.38.msi
http://www.graphviz.org/pub/graphviz/stable/windows/graphviz-2.38.zip
下载 graphviz-2.38.msi安装包,并安装,把bin路径加到系统path中。
默认安装路径:
C:\Program Files (x86)\Graphviz2.38\bin
把 C:\Program Files (x86)\Graphviz2.38\bin
加入path环境变量中。
如果是在命令行窗口使用draw_net.py,修改Path环境变量后,要重新开启命令行窗口,才能使环境变量生效。
02 使用draw_net.py画网络图
::cd %caffe_root%
cd D:\git\DeepLearning\caffe
:: 创建存放生成网络图的目录
mkdir build\x64\tmp
python python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test01.png --rankdir=BT
:: draw_net.py 的参数:
:: 第一个参数:网络模型的prototxt文件
:: 第二个参数:保存的图片路径及名字
:: 第三个参数:--rankdir=x,x有四种是LR,RL,TB,BT,表示图像生成方向。分别是从左到右、从右到左、从上到下、从下到上。默认LR。
python python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test02.png --rankdir=LR
python python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test03.png --rankdir=TB
python python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test04.png --rankdir=BT