2016-07-26 51 views
0

我對機器學習很陌生,而且我正在使用張量流進行對象檢測的iOS應用程序,我一直使用由.pb(graph.pb)文件形式的tensorflow示例提供的示例數據模型,它通過對象檢測可以很好地工作。如何將ckpt數據模型用於tensorflow iOS示例中?

但我的後端團隊給了我model2_BN.ckpt數據模型文件,我試圖研究如何使用這個文件,我不知道。 是否可以在客戶端使用ckpt文件作爲數據模型?如果是的我怎樣才能將它作爲數據模型在iOS tensorflow示例中使用?

請幫忙。 感謝

+0

回答它自己。 – coder123

回答

2

這一個從我的後端開發者:

The .ckpt is the model given by tensorflow which includes all the 
weights/parameters in the model. The .pb file stores the computational 
graph. To make tensorflow work we need both the graph and the 
parameters. There are two ways to get the graph: 
(1) use the python program that builds it in the first place (tensorflowNetworkFunctions.py). 
(2) Use a .pb file (which would have to be generated by tensorflowNetworkFunctions.py). 

.ckpt file is were all the intelligence is. 
相關問題