win7 caffe使用笔记——draw_net.py绘制caffe网络

先提供一个福利,Netscope的在线可视化工具http://ethereon.github.io/netscope/#/editor仅仅是查看网络结构的话使用这个工具,去掉#/editor也可以看到alexnet,vgg16,googlenet等网络结构。

照着ubuntu下的步骤测试成功写下过程:


1.用的是microsoft for caffe版本,pycaffe接口用的Anaconda2,给两个很好的配置博客

http://m.blog.csdn.net/article/details?id=51355143

http://blog.csdn.net/tina_ttl/article/details/51722983


2.编写bat文件,ubuntu下调用的是caffe根目录/python/draw_net.py,而windows下调根目录/Build/x64/Release/pycaffe/draw_net.py 

执行的三个参数
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第三个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。


G:
cd G:\caffe-master
SET GLOG_logtostderr=1

python  Build\x64\Release\pycaffe\draw_net.py  .\examples\Planthopper\draw_net.prototxt  examples\Planthopper\net\Planthopper_net.png
pause 

会有个错误No module named google.protobuf


3.添加python安装目录到path系统变量

添加python目录\Scripts到path系统变量

cmd命令 pip install protobuf pydot

安装Graphviz http://www.graphviz.org/Download_windows.php

bin路径加到path系统变量


4.再测试成功


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值