生成pb文件详解

本文详细介绍了如何生成.pb文件,该文件用于在Android项目中直接使用已训练的TensorFlow模型,以避免在应用中进行模型训练。内容包括训练模型的背景,导出模型的步骤,特别是如何保存训练变量的值,并使用eval()函数,以及如何新建图并利用tf.train.write_graph将模型导出为.pb文件。
摘要由CSDN通过智能技术生成

关于生成.pb的作用:

.pb文件为训练模型最终生成的文件。我们使用它的目的是为了不再android项目中进行训练而是直接套用训练的成功。

关于怎么训练模型:

本示例基于mnist手写识别项目。关于mnist手写识别训练模型的部分请参考tensorflow极客学院的中文教程中的初识mnist和深入mnist,因此关于怎么训练模型再此不在赘述。

关于怎么导出模型:

给我们导出模型的方法来自于我发现的TensorflowMnistAdroid dem。前面训练模型和极客学院的官网示例代码大同小异

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值