2017-03-15 33 views
1

我正在微調使用此的初始-v3模型花:https://github.com/tensorflow/models/tree/master/inception如何凍結inception-v3模型的檢查點目錄中的圖形?

我檢查點在目錄結果。但在目錄中我看到文件,像這樣的: -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 而我需要在目錄輸出與此類似: -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

爲此我需要做類似凍結,凍結,但需要提供output_node_names。任何人都可以指導我,inception-v3的output_node_names是什麼?另外,我需要一個可靠的方法來凍結。 tensorflow冷凍機工具可以嗎?

回答

0

我最終找到了答案。

單行答案應該是使用上游Tensorflow代碼庫中的freezer.py。查看關於如何從測試中使用該程序的示例。

您可以檢查以下鏈接示例: https://gist.githubusercontent.com/morgangiraud/249505f540a5e53a48b0c1a869d370bf/raw/6cb0b4d497925517316a92f935ce5dccb6aafd17/medium-tffreeze-1.py

+0

只是好奇,什麼是爲成立V3輸出節點的名字呢? –

+0

我在微調後嘗試了下面的腳本,但不斷收到「AssertionError」,因爲「準確性/預測」在Inception v3中不作爲輸出節點名存在。我曾嘗試用「Softmax」和其他任何合理的東西替換該字段,但它仍然不起作用。你能告訴我你是怎麼做到的嗎? –

+0

我實際上是這樣做的:https://github.com/abdasgupta/tensorflow-serving-ppc64le/blob/flowers-export/tensorflow_serving/example/inception_export.py –