1 安装pydot
$ sudo apt-get install python-pydot
2 安装graphviz
$ sudo apt-get install graphviz
3 生成
python draw_net.py ../examples/cifar10/cifar10_quick_train_test.prototxt cifa10.png
有的绘制结构图时会报错:AttrituteError:’int’ object has no attribute ‘_values’
solution: 找到python/caffe/draw.py
pooling_types_dict[layer.pooling_param.pool],
layer.type,
separator,
- layer.pooling_param.kernel_size[0] if len(layer.pooling_param.kernel_size._values) else 1,
+ layer.pooling_param.kernel_size,
separator,
- layer.pooling_param.stride[0] if len(layer.pooling_param.stride._values) else 1,
+ layer.pooling_param.stride,
separator,
- layer.pooling_param.pad[0] if len(layer.pooling_param.pad._values) else 0)
+ layer.pooling_param.pad)
else:
node_label = '"%s%s(%s)"' % (layer.name, separator, layer.type)
return node_label
注:减号即去掉这一行,加号即加上这一行
https://github.com/BVLC/caffe/commit/a2601eddf65bab54429244e350899b6d994f4f37