Tensorflow Frozen_graph to tflite for android app

问题描述 投票:0回答:2

我是TensorFlow对象检测库的新手。我有一个特定的数据集,该数据集可以产生我自己的内容,并用成千上万个jpg标记。我已经运行了该文件以从这些图像中检测对象..不管怎样。在过程结束时,我得到了Frozen_graph,并从中将model.ckpl文件导出到推理图文件夹,一切正常,并且我已经测试了model.ckpl在object_detection.ipynb文件上建立模型,效果很好。在此步骤之前,没有问题。但是,我无法理解如何将那个model.ckpl文件转换成model.tflite文件以在android studio应用上使用。

我看过很多东西like,但我不知道什么是input_tensors = [...]output_tensors = [...]我可能已经知道,但实际上是什么...

您能告诉我如何转换吗?

tensorflow tensorflow-lite
2个回答

0
投票

如果您不知道输入和输出,请使用summary_graph工具并将其作为冻结模型。在这里查看命令https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/graph_transforms#inspecting-graphs

© www.soinside.com 2019 - 2024. All rights reserved.