【神经网络与深度学习】【python开发】caffe-windows使能python接口使用draw_net.py绘制网络结构图过程
标签:【神经网络与深度学习】 【python开发】
主要是想用python绘制一下设计的网络结构图,以便可视化。因此在caffe-windows的工程配置中将python的选项设置为true,下面记录了整个成功绘图的过程。
<1> 配置CommonSettings.props中python接口
<PythonSupport>true</PythonSupport>
<PythonDir>C:\Python27\</PythonDir>
<2> 编译整个工程
在路径caffe-master\Build\x64\Release中会多了一个pycaffe文件夹,里面就是针对python-caffe的接口
<3> 给python安装一系列工具
PS:不知道啥时候我的windows上安装了pip和easy_install,这两个工具简直是逆天啊!easy_install用了一次安装scikit_image结果不知道为何把我的电脑卡死了,我强制关机了一下,还好系统没问题。所以,以后都比较倾向用pip:
(1) 首先是 protobuf:
pip install protobuf
如果 版本不够高的话,可能还是会出现importerror的错误,可以使用升级一下:
pip install protobuf --upgrade
(2) 然后安装pydot
pip install pydot
(3) 安装GraphViz
pip install GraphViz
(4) 安装scikit_image
pip install scikit-image //这个我的电脑报错了
后来在这个问题上费了好大一番功夫,python只是好几个月前稍微学了一下,当时掌握的如何进行配置的东西基本上全忘了。好在终于没有放弃,然后找到了在windows上的一些对应的extensions:
http://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy
从里面下载了scikit_image-0.12.3-cp27-cp27m-win_amd64.whl(说明对应的是python版本号与操作系统的x64还是win32)。
然后直接:
pip install scikit_image-0.12.3-cp27-cp27m-win_amd64.whl
(5) 同样地安装scipy
pip install scipy-0.18.0-cp27-cp27m-win_amd64.whl
(6) 安装numpy-1.11.1+mkl
pip install numpy-1.11.1+mkl-cp27-cp27m-win_amd64.whl
经过上面一系列的安装和配置,最后将deploy.prototxt复制到Release文件夹下,cmd到该路径下,然后执行如下命令:
python draw_net.py deploy.prototxt net.bmp --rankdir=BT
draw_net.py执行的时候带三个参数
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第三个参数:–rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。
2016-9-17 16:58
张朋艺 pyZhangBIT2010@126.com