2017-03-15 9 views
1

私は、これを使用して、初期化-v3モデルの花を微調整しています。https://github.com/tensorflow/models/tree/master/inceptioninception-v3モデルのチェックポイントディレクトリからグラフをフリーズする方法はありますか?

私は結果をディレクトリにチェックポイントしました。しかし、ディレクトリに私はこれらのようなファイルを参照してください。output_node_namesを提供するために、私は凍結のような何かをする必要があり、そのために -rw-r----- 1 107456 5000 223 Mar 2 2016 README.txt -rw-r----- 1 107456 5000 43 Mar 2 2016 checkpoint -rw-r----- 1 107456 5000 434903494 Mar 15 2016 model.ckpt-157585

が、凍結ニーズ: -rw-r--r-- 1 root root 389908432 Mar 15 21:46 model.ckpt-0.data-00000-of-00001 -rw-r--r-- 1 root root 72680 Mar 15 21:46 model.ckpt-0.index -rw-r--r-- 1 root root 15189794 Mar 15 21:47 model.ckpt-0.meta -rw-r--r-- 1 root root 135185788 Mar 15 22:36 events.out.tfevents.1489594533.f7d5defbed64 -rw-r--r-- 1 root root 72680 Mar 15 22:37 model.ckpt-4999.index -rw-r--r-- 1 root root 389908432 Mar 15 22:37 model.ckpt-4999.data-00000-of-00001 -rw-r--r-- 1 root root 15189794 Mar 15 22:38 model.ckpt-4999.meta -rw-r--r-- 1 root root 130 Mar 15 22:49 checkpoint 私はこのようなディレクトリでの出力を必要とするのに対し。誰でも私を導くことができます、開始-v3のoutput_node_namesは何ですか? また、フリーズするには信頼できる方法が必要です。テンソルフローフリーザーツールはこれで大丈夫ですか?

答えて

0

最終的に答えが見つかりました。

1行の答えは、上流Tensorflowコードベースで使用可能なfreezer.pyを使用することです。テストからそのプログラムを使用する方法の例を参照してください。

あなたはサンプルのため、以下のリンクを確認することができ: https://gist.githubusercontent.com/morgangiraud/249505f540a5e53a48b0c1a869d370bf/raw/6cb0b4d497925517316a92f935ce5dccb6aafd17/medium-tffreeze-1.py

+0

興味があるだけに、インセプションv3の出力ノード名は何でしたか? –

+0

Inception v3の出力ノード名として「精度/予測」が存在しないため、微調整後にスクリプトを試しましたが、「AssertionError」が継続していました。私はそのフィールドを「Softmax」と置き換えようとしましたが、それ以外は妥当なものですが、それでも動作しません。あなたはそれをどうやって教えてくれますか? –

+0

私は実際にこれを行いました:https://github.com/abdasgupta/tensorflow-serving-ppc64le/blob/flowers-export/tensorflow_serving/example/inception_export.py –

関連する問題