caffe2网络结构可视化中pydot和graphviz的使用_pydot和graphviz的区别-优快云博客

本文链接：https://blog.youkuaiyun.com/Hia_Hia_Hia/article/details/83040046

本文介绍两种Caffe2网络结构的可视化方法：对于小规模网络，可通过输出网络结构到文档进行查看；对于大规模网络，利用graphviz和pydot包结合Caffe2的net_drawer模块，将网络结构以图形方式呈现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用过caffe的都知道，caffe的网络结构可以用NetScope可视化，但caffe2目前没有这么直接的工具，为了可视化网络结构，有两个思路。

1、对于层数较少的网络，可以输出网络结构，命令行不方便查看的话可以输出到文档

f=open('/home/usr/proto.txt','w') 
print(net.Proto(),file=f)
f.close()

2、对于层数较多的网络，只是输出成文字显然很不方便

（1）先安装需要的包

pip install graphviz

sudo apt-get install graphviz

pip install pydot

用别的安装方法可能一开始也不会报错，但后面可能会出现一些问题，比如我的代码就在调用pydot中write_png方法时卡了一天一夜都动不了，按上面的安装方法就能正常运行。

https://blog.youkuaiyun.com/sinat_37998852/article/details/80507536

（2）再用以下代码让网络结构以图形的方式呈现

显示全部参数和operators

from caffe2.python import net_drawer
from IPython import display
graph=net_drawer.GetPydotGraph(net.Proto(),rankdir="LR") #LR表示从左到右，也可选择TB —— 从上到下
display.Image(graph.create_png(),width=800) # 显示图片
graph.write_png('proto.png')) # 保存图片

只显示operators

from caffe2.python import net_drawer
from IPython import display
graph=net_drawer.GetPydotGraphMinimal(net.Proto(),rankdir="LR", minimal_dependency=True)
display.Image(graph.create_png(),width=800) #显示图片
graph.write_png('proto.png')) # 保存图片