如何在 .pb 文件中找到输出节点的确切名称？

Question

我试图冻结一个 pb 文件以便在 OpenVino 中使用。为了冻结，我需要知道输出节点名称。为此，我尝试加载 pb 文件并读取输出名称，但是出现错误。然后我尝试从模型摘要中获取输出名称，它是 dense_7（密集）。

我按照https://docs.openvinotoolkit.org/latest/_docs_MO_DG_prepare_model_convert_model_Convert_Model_From_TensorFlow.html#freeze-the-tensorflow-model中的命令冻结了模型。但是当我尝试将输出节点名称设为 dense_7（密集）时，出现以下错误：

AssertionError: dense_7 (Dense) is not in graph

我附上模型摘要。

有没有办法读取正确的输出名称？

Answer 1

• “模型优化器常见问题解答”页面中的问题 26 解决了您遇到的错误。参见 https://docs.openvinotoolkit.org/2020.3/_docs_MO_DG_prepare_model_Model_Optimizer_FAQ.html

• 我建议您尝试在 Github 中共享的以下潜在方法，可从以下链接获得：

  https://github.com/tensorflow/tensorflow/issues/3986#issuecomment-304619868 https://github.com/tensorflow/tensorflow/issues/3986#issuecomment-368282326 https://github.com/tensorflow/tensorflow/issues/3986#issuecomment-470083664 https://github.com/tensorflow/tensorflow/issues/3986#issuecomment-508897579

• 如果这些对您有用，请告诉我们。