如何将ckpt数据模型用于tensorflow iOS示例?

时间:2016-07-26 13:37:41

标签: ios machine-learning tensorflow

我对机器学习很安静,我正在使用tensorflow进行对象检测的iOS应用程序,我一直在使用由.pb(graph.pb)文件形式的tensorflow示例提供的示例数据模型这对于物体检测效果很好。

但我的后端团队给了我model2_BN.ckpt数据模型文件,我试图研究如何使用这个文件,我不知道。  是否可以在客户端使用ckpt文件作为数据模型?如果是,我如何在iOS张量流示例中将其用作数据模型?

请帮忙。 感谢

1 个答案:

答案 0 :(得分: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.