当前位置: 代码迷 >> 综合 >> TensorFlow:从模型文件看它的结构
  详细解决方案

TensorFlow:从模型文件看它的结构

热度:66   发布时间:2023-12-08 23:11:27.0

meta文件读取方式

import tensorflow as tf

from tensorflow.python.platform import gfile #这是从文件格式的meta文件加载模型

graph = tf.get_default_graph()

graphdef = graph.as_graph_def()

_ = tf.train.import_meta_graph("***/model.meta")

summary_write = tf.summary.FileWriter("***/summary/" , graph)

 

pb文件读取方式

import tensorflow as tf

from tensorflow.python.platform import gfile #这是从文件格式的meta文件加载模型

graph = tf.get_default_graph()

graphdef = graph.as_graph_def()

graphdef.ParseFromString(gfile.FastGFile("/****/model.pb", "rb").read())

_ = tf.import_graph_def(graphdef, name="")

summary_write = tf.summary.FileWriter("****/summary3/" , graph)

 

然后再启动tensorboard:

tensorboard --logdir /data/TensorFlowAndroidMNIST/logdir --host 你的ip --port 你端口(默认6006)

tensorboard --logdir "***/summary/"

tensorboard --logdir "***/summary2/"

 

 

 

 

  相关解决方案