从Tensorflow模型文件中解析并显示网络结构图(pb模型篇)1 Tensor对象与Operation对象2 提取pb文件中的网络结构图3 绘制网络结构4 测试模型显示5 源码地址
Tensorflow官方提供的Tensorboard可以可视化神经网络结构图,但是说实话,我几乎从来不用。主要是因为Tensorboard中查看到的图结构太混乱了,包含了网络中所有的计算节点(读取数据节点、网络节点、loss计算节点等等)。更可怕的是,如果一个计算节点是由多个基础计算(如加减乘除等)构成,那么在Tensorboard中会将基础计算节点显示而不是作为一个整体显示(典型的如Squeez...