我正在尝试将 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)