caffe network 可视化

本博文介绍caffe network可视化，非常的简单
首先，安装必须的dependencies：

$ pip install pydot
$ sudo apt-get install graphviz libgraphviz-dev
$ pip install pygraphviz

在这假设caffe已经安装好了，下面给一个简单的例子，在test.prototxt文件中创建一个网络:

name:”myconvolution”
input: “data”
input_dim: 1
input_dim: 1
input_dim: 256
input_dim: 256
layer {
  name: “conv”
  type: “Convolution”
  bottom: “data”
  top: “conv”
  convolution_param {
    num_output: 10
    kernel_size: 3
    stride: 1
    }
    weight_filler{
      type: “gaussian”
      std: 0.01
    }
    bias_filler {
      type: “constant”
      value: 0
    }
  }
}

安装dependencies后,运行

$ python /path/to/caffe/python/draw_net.py  t