有没有其他转换到tflite的方法？

Question

我遵循了这个程序：

https://kharbari.com/how-to-convert-pb-to-tflite-format/

但我面临这个错误：

“名称‘prefix/MultipleGridAnchorGenerator/assert_equal_1/Assert/Assert:0’指的是一个不存在的张量。操作‘prefix/MultipleGridAnchorGenerator/assert_equal_1/Assert/Assert’存在但只有0个输出。”

我的代码是：

import tensorflow as tf

def load_graph(frozen_graph_filename):
with tf.compat.v2.io.gfile.GFile(frozen_graph_filename, "rb") as f:
    graph_def = tf.compat.v1.GraphDef()
    graph_def.ParseFromString(f.read())

with tf.Graph().as_default() as graph:
    tf.import_graph_def(graph_def, name="prefix")
return graph


if __name__ == '__main__':
graph = load_graph('/content/123trained.pb')
for op in graph.get_operations():
    abc = graph.get_tensor_by_name(op.name + ":0")
    print(abc)

有人可以帮我调试吗？

如果有其他程序，请告诉我！！

提前致谢！

Answer 1

TF.Assert 操作没有任何输出张量。 但是，您似乎正在尝试在“abc = graph.get_tensor_by_name(op.name + ":0")”中访问它