2017-07-20 16 views
1

私はペット検出器のチュートリアルに従いました。私は "export_inference_graph.py"を使ってモデルをエクスポートしました。"tensorflowオブジェクト検出apiペット検出器"グラフをフリーズ

しかし、私は、提供された "freeze_graph.py"を使ってグラフをフリーズしようとしましたが、今は--output_node_namesが何を使うべきかを確かめます。

私は自分のモデルを訓練するときにどのように使うべきかを知っている人がいますか?

答えて

0

output_node_namesに使用するものを見つけるには、graph.pbtxtファイルをチェックアウトしてください。この場合、Softmaxでした

0

export_inference_graph.pyの出力は現在すでにフリーズグラフになっているはずですので、もう一度フリーズしようとすると冗長になります。

結果のフリーズしたグラフの入力ノード名と出力ノード名については、私たちが提供するjupyter notebookをチェックアウトしてください。

+0

ありがとうございます。私は凍ったグラフで提供されたメモ帳を実行しましたが、現在このエラーが発生しています: KeyError: "image_tensor:0"という名前は存在しないTensorを指します。グラフ。 –

+0

detect_graph.get_operations()からエクスポートされたすべてのオペレーション名のリストhttps://pastebin.com/3XvMZrLE –

+0

私は類似の操作がないので、何かが深刻に間違っていると仮定しています。私はチュートリアルに従って、自分の画像とlabelImgを使って注釈を付けました。 –

関連する問題