問題タブ [attention-model]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
python - OpenCV-DNN TensorFlow の関数 'ReadTFNetParamsFromTextFileOrDie' で GraphDef ファイルを解析できません
Attention-OCRモデルを OpenCV-DNN でラップして、推論時間を増やしたいと考えています。公式TFモデルレポのTFコードを使用しています。
TF モデルを OpenCV-DNN でラップするには、このコードを参照しています。TF モデルを読み取るにはcv2.dnn.readNetFromTensorflow()
、「凍結グラフ」と「グラフ構造」が必要です。
このコード スニペットを使用して、メタ チェックポイント ファイルから構造をインポートし、グラフ構造をファイルに保存し.pbtxt
ます。
グラフを固定するためのコードは次のとおりです。
最終的なコードでは、関数内でpbtxt
およびpb
ファイルを使用しcv2.dnn.readNetFromTensorflow()
ます。
発生したエラーは次のとおりです。
注:出力ノード名は、以下を使用して生成されたグラフ内のテンソルのリストを見て手動で設定されます。
SO コミュニティから提供されたヘルプに感謝します。