使用GraphViz画caffe的网络结构图

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/happynear/article/details/45440709

  caffe的python接口中有一个很不错的功能:画网络结构图,虽然画得并不好看,但可以给人一种直观的感受。

一、准备

  首先caffe的python接口当然是必备的了,还没有生成python接口的同学可以参照我的上一篇博客来生成。
  然后是需要安装protobuf的python接口,可以参照这篇博客进行安装,安装过程比较简单,就不赘述了。
  安装GraphViz:http://www.graphviz.org/
  安装pydot,可以使用pip install pydot来安装,但最近这个功能好像不太稳定(GFW,你懂的),大家可以从https://github.com/nlhepler/pydot 下载,这里需要修改一下pydot/__init__.py文件中,第596行:

path = os.path.join(os.environ['PROGRAMFILES'], 'ATT', 'GraphViz', 'bin')

  改为:

path = r"C:\Program Files (x86)\Graphviz2.37\bin"

  同样,下边的那个path也改为r”C:\Program Files (x86)\Graphviz2.37\bin”,如果你将GraphViz安装在别的位置,那这个目录也要跟着改变。

二、生成网络结构图

  把环境配置好之后,就可以在命令行中输入
   python draw_net.py ..\examples\mnist\lenet_train_test.prototxt 1.bmp
   就可以看到在当前目录下生成了一张图片:
   lenet的网络结构

猜你喜欢

转载自blog.csdn.net/happynear/article/details/45440709