先提供一个福利,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.再测试成功