【问题标题】:How to use ckpt data model into tensorflow iOS example?如何在 tensorflow iOS 示例中使用 ckpt 数据模型?
【发布时间】:2016-07-29 22:02:15
【问题描述】:
我对机器学习很陌生,我正在开发 iOS 应用程序以使用 tensorflow 进行对象检测,我一直在使用 tensorflow 示例以 .pb (graph.pb) 文件的形式提供的示例数据模型与对象检测配合得很好。
但是我的后端团队给了我model2_BN.ckpt作为数据模型文件,我试图研究如何使用这个文件,我不知道。
是否可以将客户端的 ckpt 文件用作数据模型?如果是,如何在 iOS tensorflow 示例中使用它作为数据模型?
请帮忙。
谢谢
【问题讨论】:
标签:
ios
machine-learning
tensorflow
【解决方案1】:
这个来自我的后端开发人员:
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.