caffe的python编译模块提供了绘制net结构的功能,可实现对网络结构的绘制,这极大的方便了对网络结构的理解和分析,这里介绍下绘制的过程。
一、准备工作:
1、确保caffe的python模块已经编译和配置。
具体可参考【caffe】windows下caffe+vs2013+python2.7+cuda8.0+cmake3.8编译与配置
2、打开命令行cmd窗口,进入python安装目录下的Scripts目录(这个目录需要添加到环境变量里,否者pip安装命令无法使用)。
3、安装protobuf模块:pip install protobuf。
4、安装graphviz模块:pip install graphviz。
5、安装graphviz软件:下载graphviz-2.38.msi