在預測中使用TensorFlow對象檢測API模型

Question

我已經使用TensorFlow object detection API從零開始訓練SSD初始模型。評估腳本顯示模型已經學到了一些東西，現在我想使用模型。

我已經看過物體檢測ipynb，它可以將單個圖像傳送給訓練有素的模型。但是，這是用於MobileNet的SSD。我已經使用了下面一行（加載元圖之後）來打印我訓練過的TensorFlow模型的張量名稱。

print([str(op.name) for op in tf.get_default_graph().get_operations()]) 

但它不包含與ipynb中相同的輸入或輸出張量名稱。我也搜遍了代碼，但許多功能指向對方，很難找到我想要的東西。

如何找到我需要的張量名稱？還是有另一種我不知道的方法？

Answer 1

要使用圖表，您需要凍結/導出它，使用this provided script。生成的.pb文件將包含您需要的節點。我不知道爲什麼它是這樣組織的，但它是。