小编Ale*_*scu的帖子

如何将 TensorFlow 模型导出为 .tflite 文件?

背景资料:

我编写了一个非常类似于TensorFlow 提供的预制虹膜分类模型的 TensorFlow模型。差异相对较小:

  • 我正在对足球运动进行分类,而不是虹膜种类。
  • 我有 10 个特征和一个标签,而不是 4 个特征和一个标签。
  • 我有 5 种不同的练习,而不是 3 种鸢尾花。
  • 我的 trainData 包含大约 3500 行,而不仅仅是 120。
  • 我的 testData 包含大约 330 行,而不仅仅是 30。
  • 我正在使用 n_classes=6 而不是 3 的 DNN 分类器。

我现在想将模型导出为.tflite文件。但根据TensorFlow 开发人员指南,我需要先将模型导出到tf.GraphDef文件,然后将其冻结,然后才能进行转换。但是, TensorFlow 提供的从自定义模型创建文件的教程.pb似乎仅针对图像分类模型进行了优化。

题:

那么如何将像虹膜分类示例模型这样的模型转换成.tflite文件呢?有没有更简单、更直接的方法来做到这一点,而不必将其导出到.pb文件,然后将其冻结等等?基于虹膜分类代码的示例或指向更明确教程的链接将非常有用!


其他信息:

  • 操作系统:macOS 10.13.4 High Sierra
  • TensorFlow 版本:1.8.0
  • Python 版本:3.6.4
  • 使用 PyCharm 社区 2018.1.3

代码:

输入以下命令可以克隆虹膜分类代码:

git clone https://github.com/tensorflow/models

但如果你不想下载整个包,这里是:

这是名为的分类器文件premade_estimator.py

    # …
Run Code Online (Sandbox Code Playgroud)

python machine-learning pycharm tensorflow tensorflow-lite

6
推荐指数
1
解决办法
1万
查看次数