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

我需要.tflite文件来使用我在android中训练过的模型

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

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

我听说我可以学习输入数组和 从tensorboard输出_数组值，但它们不显示在其中。我使用以下代码：

tensorboard --logdir=object_detection\out\

我的tensorboard屏幕中存在以下错误：

“我的输出”文件夹如下所示：

---

对此我能做些什么？

您可以使用汇总图来检查该图。 这可以帮助您解决问题

---

检查输入数组和输出数组。在冻结的\u推断\u graph.pb文件夹所在的位置创建一个python文件，然后将此代码粘贴到py中

import tensorflow as tf
gf = tf.GraphDef()
m_file = open('tflite_graph.pb','rb')
gf.ParseFromString(m_file.read())

with open('somefile.txt', 'a') as the_file:
    for n in gf.node:
        the_file.write(n.name+'\n')

file = open('somefile.txt','r')
data = file.readlines()
print ("output name = ")
print (data[len(data)-1])

print ("Input name = ")
file.seek ( 0 )
print (file.readline())

运行文件后，它将显示输入数组和输出数组

详情请访问：