背景资料:
我编写了一个非常类似于TensorFlow 提供的预制虹膜分类模型的 TensorFlow模型。差异相对较小:
我现在想将模型导出为.tflite文件。但根据TensorFlow 开发人员指南,我需要先将模型导出到tf.GraphDef文件,然后将其冻结,然后才能进行转换。但是, TensorFlow 提供的从自定义模型创建文件的教程.pb似乎仅针对图像分类模型进行了优化。
题:
那么如何将像虹膜分类示例模型这样的模型转换成.tflite文件呢?有没有更简单、更直接的方法来做到这一点,而不必将其导出到.pb文件,然后将其冻结等等?基于虹膜分类代码的示例或指向更明确教程的链接将非常有用!
其他信息:
代码:
输入以下命令可以克隆虹膜分类代码:
git clone https://github.com/tensorflow/models
但如果你不想下载整个包,这里是:
这是名为的分类器文件premade_estimator.py:
# …Run Code Online (Sandbox Code Playgroud)