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ファイルには、必要なノードが含まれます。私はそれがなぜそんなに組織化されているのかわかりませんが、それはそうです。