本博文介绍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 test.prototxt test.png
在我的电脑上是:
$ python ~/caffe/python/draw_net.py test.prototxt test.png
可视化后的结果如下图所示:
在这给出另外一个例子,可视化caffe examples中的mnist中的lenet.prototxt 网络:
$ python ~/caffe/python/draw_net.py ~/caffe/examples/mnist/lenet.prototxt lenet.png
lenet 网络 可视化后的结果如下图:
可视化后有助于你更好的了解网络的结构,如果感兴趣可以尝试其他的网络.