caffe绘制网络结构图

最新推荐文章于 2024-04-23 00:55:51 发布

薪升贷农名工

最新推荐文章于 2024-04-23 00:55:51 发布

阅读量915

点赞数

文章标签： python 结构网络 graphviz

本文链接：https://blog.csdn.net/z13653662052/article/details/70948236

版权

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