小编Som*_*hah的帖子

如何从 Tensorflow 中的冻结模型(pb 文件)中找到 output_node_names?

我正在尝试将 freeze_model.pb 转换为 TensorFlow JS 兼容(.pb)文件,该文件基于 Tensorflow 的 SSD Mobilenet V2 COCO 预训练模型。我陷入了如何获取使用tensorflowjs_converter 时所需的output_node_names 参数的困境。如何知道输出节点名称?

我尝试使用下面的 Python 脚本获取操作名称,但无法理解哪一个是输出节点。

def load_graph(model_file):
  graph = tf.Graph()
  graph_def = tf.GraphDef()

  with open(model_file, "rb") as f:
    graph_def.ParseFromString(f.read())
  with graph.as_default():
    tf.import_graph_def(graph_def)

  return graph

graph = load_graph('frozen_model.pb')
ops = graph.get_operations()
Run Code Online (Sandbox Code Playgroud)

python tensorflow

3
推荐指数
1
解决办法
3913
查看次数

标签 统计

python ×1

tensorflow ×1