caffe 03 win10 使用 draw_net.py 画网络图

来源:互联网 发布:网络西瓜是什么意思啊 编辑:程序博客网 时间:2024/05/22 05:22

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\tmppython 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=LRpython python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test03.png --rankdir=TBpython python\draw_net.py examples/mnist/lenet_train_test.prototxt build\x64\tmp\lenet_test04.png --rankdir=BT

生成的网络图

0 0