2017-07-06 67 views
-1

我使用提供的腳本here從頭開始培訓初始模型。 訓練的輸出是這些文件:將model.ckpt轉換爲可用模型

checkpoint 
events.out.tfevents.1499334145.fdbf-Dell 
model.ckpt-5000.data-00000-of-00001 
model.ckpt-5000.index 
model.ckpt-5000.meta 
... 
model.ckpt-25000.data-00000-of-00001 
model.ckpt-25000.index 
model.ckpt-25000.meta 

是否有人有一個腳本,這些文件的東西,我可以用我的圖片分類轉換?我已經嘗試修改inception_train.py文件輸出graph.pb,但沒有任何反應...

任何幫助將不勝感激,謝謝!

+0

這裏有很多關於這個問題,你沒有搜索太久...... – gdelab

+1

有到處都是不完整和部分的答案。我正在尋找一個簡單但完整的腳本,可以幫助我使用這些文件。例如,從頭開始訓練的「官方」指南確實解釋瞭如何訓練以及如何評估模型,但並未解釋如何使用它來對其他圖像進行分類......爲什麼? –

回答

0

如何直接使用您的檢查點解釋爲here

要創建一個.pb文件,您必須凍結圖形,如解釋here

要創建初始.pb(或.pbtxt)需要freeze_graph文件,你可以使用SO tf.train.write_graph()

+0

謝謝。要凍結圖形,它需要一個graph.pbtxt文件,我找不到任何地方......它應該在哪裏? –

+0

我已經更新了我的答案 – gdelab

+0

非常感謝你! –