【神经网络与深度学习】【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) 安装ＧraphViz

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