我无法将.pb文件转换为.tflite

Question

我需要.tflite文件才能使用我在android中训练的模型。

我将使用此代码，但我不知道输入数组和输出数组的值：

<script src="https://code.jquery.com/jquery-3.4.1.slim.min.js" integrity="sha256-pasqAKBDmFT4eHoN2ndd6lN370kFiGUFyTiUHWhU7k8=" crossorigin="anonymous"></script>

<button type="button" id="displayMe" onclick="displayFunction()">
Click me to add the word "Display" to the page.</button>

<p id="demo"></p>

听说我可以学习input_arrays和 张量板的output_arrays值，但未在其中显示。我使用以下代码：

  tflite_convert \ 
--output_file=object_detection\outputtflite  \ 
--graph_def_file=object_detection\out\frozen_inference_graph.pb\ 
--input_arrays=input\ 
--output_arrays=output

我的张量板屏幕中存在以下错误：

我的out文件夹如下：

我该怎么办？

Answer 1

您可以使用summary_graph来检查图形。 这可以帮助您解决问题 https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/graph_transforms#inspecting-graphs

Answer 2

检查输入数组和输出数组。在Frozen_inference_graph.pb文件夹所在的位置创建一个python文件，然后将此代码粘贴到py

<span data-mo-top-fade-duration="600">Title 1</span>
<span data-mo-bottom-fade-duration="600">Title 2</span>
<span data-mo-right-fade-duration="600">Title 3</span>
<span data-mo-left-fade-duration="600">Title 4</span>
<span>somethingElse</span>

运行文件后，它将显示input_arrays和output_arrays。

进一步的信息检查：How to convert .pb to TFLite format?