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

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

生成的网络图

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值