python - 对 tensorflow 诗人中其他架构的评估.py 中的修改

Question

在tensorflow-for-poets-2/scripts/evaluate.py中有一个用于测量模型性能的脚本。该脚本最初是为 mobilenet 模型编写的，但可以针对其他架构进行相应的修改。我应该在第41行写什么

logits = graph.get_tensor_by_name("final_training_ops/Wx_plus_b/add:0"). 

我收到`KeyError：

名称 final_training_ops/Wx_plus_b/add:0' 指的是不存在的张量。图中不存在“final_training_ops/Wx_plus_b/add”操作。

标签： pythontensorflowconvolutional-neural-network

Answer 1

我自己想通了。给定一个 .pb 文件，我们可以使用此代码查看冻结的 .pb TensorFlow 图中的操作。因此，我们需要根据我们的 .pb 文件将第41行替换为最后一个操作。