caffe绘制网络结构图

来源：互联网发布：efd软件下载编辑：程序博客网时间：2024/04/29 06:06

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

这里写图片描述

0 0