【神经网络与深度学习】【python开发】caffe-windows使能python接口使用draw_net.py绘制网络结构图过程

【神经网络与深度学习】【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。

【神经网络与深度学习】【python开发】caffe-windows使能python接口使用draw_net.py绘制网络结构图过程


2016-9-17 16:58

张朋艺 pyZhangBIT2010@126.com

上一篇:深度学习 + OpenCV,Python实现实时视频目标检测


下一篇:使用Iterator迭代器循环集合