caffe 使能python接口使用draw_net.py绘制网络结构图过程

来源:互联网 发布:手机淘宝支付宝登录 编辑:程序博客网 时间:2024/04/27 16:34

前一章很不容易把python接口配置好了,接下来做测试,绘制网络结构图。又遇到各种奇葩问题。

绘制命令如下:

python draw_net.py deploy.prototxt net.bmp --rankdir=BT

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

记录如下:

1.cmd到....\caffe-master\Build\x64\Release\pycaffe目录下,可以看到有我们需要的draw_net.py,然后可以将models\bvlc_googlenet\train_val.prototxt拷贝到这个目录下,输入上述命令,会报错Import Error:No module named protobuf.,然后看到外国友人也问了这一帖子,网址链接http://stackoverflow.com/questions/38680593/importerror-no-module-named-google-protobuf。出现错误的原因When pip tells you that you already have protobuf, but PyCharm (or other) tells you that you don't have it, it means that pip and PyCharm are using a different Python interpreter. 参照这个人讲的


他是在linux下编的,然后百度了一下windows virtualenv.找到这个http://www.2cto.com/os/201411/355798.html,就是在windows配置python的虚拟机环境。

2.按照上面方式配置好windows下的python环境后,接下来还是按照http://blog.csdn.net/lg1259156776/article/details/52563889这个网址提供的在虚拟机环境下,把那些包重新下载一遍,下载完成后重复第一步一开始的操作,这时会报dot.exe not found in path,这时又去解决这个错误,看这个网址的http://stackoverflow.com/questions/13987353/python-valueerror-program-dot-not-found-in-path

先去这个网址http://www.graphviz.org/Download_windows.php将graphviz-2.38.msi下下来,进行安装,可以发现它的bin目录下有这个我们需要的dot.exe,将这个bin目录添加到系统的环境变量中去即可,再pip install  pygraphviz即可。

3.最后重启电脑,使环境变量生效,再次运行即可,得到我们想要看到的网格结构图。


大功告成,网格图还是很he人的哈~~~


0 0
原创粉丝点击