Caffe在windows下用python绘制网络结构图

本篇文章是关于caffe如何在windows下用python绘制网络结构图。

需要准备的东西有以下：

①windows版本下的caffe

②Anaconda（python2.7 64位版本）32位版本的可以自行到官网下载

③graphviz-2.38.msi

准备的东西都已经放在了百度网盘中。

第一步： 配置安装caffe。（这里参考上一篇文章有关如何在windows下配置安装caffe）

强调一下：在配置安装caffe时，在“”CommonSettings.props“文件中一定要这样选择

”，并且把Anaconda的存放的地址写上，例如我的是。

第二步：在完整caffe的安装配置后。在.../Build/x64/DeBug里面有产生文件夹 pycaffe。

。

打开pycaffe这个文件夹，里面有这些内容，把红圈圈住的caffe文件夹复制一下。

然后，把这个文件夹粘贴到下面的位置。

安装完Anaconda后，找到Anaconda的位置，打开Lib文件夹。

。

打开Lib文件夹后，找到site-packages文件夹，把刚刚复制的文件夹，粘贴在这个文件夹中。

第三步：打开Anaconda文件夹，运行python.exe

然后在命令行中输入 import caffe，回车后等待运行。

一般会有以下这种错误，提示No module named caffe，这个一般是刚刚没有把caffe文件夹复制到site-packages文件夹中。

第四步：安装graphviz-2.38.msi，新建一个文件夹名为graphviz， 把图中的文件都放在这个文件夹中，

注意：路径中的不能含有空格，否则会出现后面 dot.exe not found的错误。

安装完之后，添加系统环境变量。具体如下

打开我的电脑--->右键-->属性-->高级系统设置-->高级-->环境变量

点击编辑后，按新建。

然后把graphviz文件夹里面的bin添加到路径中。如下图这个样子。

最后：新建一个记事本。里面敲入以下内容：

python E:\KingSam\caffe\caffe-windows（2）\caffe-windows\Build\x64\Debug\pycaffe\draw_net.py cifar10_quick_train_test.prototxt aa.png
pause

保存记事本内容，然后将格式从txt更改成bat，双击运行后便产生了网络结构图。

最后：在import caffe的时候可能会出现各种各样的问题，例如还缺少一些model之类的，这些问题希望读者都评论区留下，我会尽力为大家解决！