无法检测节点名称并冻结微调的inceptionv3模型

时间:2017-10-02 16:28:45

标签: python-2.7 tensorflow

虽然之前已经问过这个问题,但我想问这个问题,因为我是tensorflow inceptionv3模型的新手,我无法冻结我的模型并生成Protobuf文件(output_graph.pb)。

我有一个微调的初始v3模型,带有自定义数据集,并获得数据,元文件和索引文件。根据要冻结模型 https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-serve-it-with-a-python-api-d4f3596b3adc 我需要知道outputnode名称。我试着用这个 https://github.com/tensorflow/models/issues/544表示节点,并将tower_0 / logits / logits / xw_plus_b:0作为输出节点。但是,当我使用此节点名称冻结模型时,我收到错误 AssertionError:tower_0 / logits / logits / xw_plus_b:0不在图中

那么,有没有其他方法可以找到输出节点名称?如果是这样,我可以使用元流链接模型来冻结我的模型并获得output_graph.pb吗?

1 个答案:

答案 0 :(得分:1)

找到了节点名称。我试过,tower_0 / logits / predictions并且它有效