Tensorflow：事前にトレーニングされたCNNの特定のレイヤーからイメージ機能を抽出するにはどうすればよいですか？

Question

私は訓練されたCNNモデルを.pbファイルとして持っています。私はモデルをロードし、すべての画像の最後のレイヤーから最終ベクトルを抽出することができます。今私は特定のレイヤーから来て、最終的なイメージではないベクトルを抽出したいと思います。私はモデルを読み込むためにimport_graph_def関数を使用しています。.pbファイルが大きいため、ファイルを開くことができないため、レイヤの名前はわかりません。

私が望むレイヤーまでベクトルを取得するために、モデルの一部を実行して全体を実行する方法はありますか？

Answer 1

コメントにtf.Graph.get_operations（）を使用するPeter Hawkins氏以外のアプローチは、テンソルボードを使用して、抽出したいレイヤの名前を見つけることです。

そこから、あなたはちょうどあなたがしたい方の特徴アウト抽出すること

graph.get_tensor_by_name("import/layer_name") 

を使用することができます。