Tensorflow的pb文件分析

阅读量：

使用Tensorboard分析pb文件有两种方法：

方法一：

通过加载pb文件来重建计算图结构

借助Tensorboard工具分析计算图的架构组成

方法二

可采用tensorflow内置的tools模块中提供的import_pb_to_tensorboard.py脚本进行操作，不过需要注意的是，该工具在linux系统中默认未安装（windows系统则默认集成），如需使用可前往指定链接下载[https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/tools]

方法一

1.从pb文件中恢复计算图

复制代码

    import tensorflow as tf
    
    model = 'model.pb' #请将这里的pb文件路径改为自己的
    graph = tf.get_default_graph()
    graph_def = graph.as_graph_def()
    graph_def.ParseFromString(tf.gfile.FastGFile(model, 'rb').read())
    tf.import_graph_def(graph_def, name='graph')

全部评论 (0)

还没有任何评论哟~