web-dev-qa-db-ja.com

TensorFlow用にC ++でチェックポイントと推論をロードする方法は?

python apiのTensorflowがあり、次のチェックポイントモデルファイルを取得しました:

model.ckpt-17763.data-00000-of-00001
model.ckpt-17763.index
model.ckpt-17763.meta

しかし、本番環境に統合するときは、C/C++共有ライブラリ(.soファイル)が必要です。したがって、これらのモデルファイルと推論をC++コードでロードし、共有ライブラリにコンパイルする必要があります。これを行うためのチュートリアルやサンプルはありますか?

7
tidy

here の指示に従って、グラフをロードして使用するc ++コードを記述できます。

ファイル ここ を使用して、TFリポジトリの外部でtensorflowを使用してCmakeプロジェクトを作成し、ライブラリをコンパイルできます。

ただし、.soの横に.ckptファイルが必要なため、その中にインターゲートする方法がわかりません。

SOにはそれについて多くの質問があり、いくつかのチュートリアルがあります( この回答 で引用されている2つを参照)が、テンソルフローは急速に進化しているため、すぐに時代遅れになり、常に少し苦労しますそれを適切に機能させるために(完全に実行可能であり、簡単になりますが)。

7
gdelab