列出TFLite模型的操作

Question

尝试列出TFLite模型的操作时遇到麻烦。我知道可以通过冻结图来列出操作，但是TFLite .tflite模型又如何呢？可以列出操作吗？

Answer 1

如TensorFlow Lite docs中所述，您需要使用tf.lite.Interpreter来解析.tflite模型。

# Load TFLite model and allocate tensors.
interpreter = tf.lite.Interpreter(model_path="converted_model.tflite")
interpreter.allocate_tensors()

然后使用get_tensor_details方法获取张量列表。

interpreter.get_tensor_details()

根据文档，

  

获取具有有效张量详细信息的每个张量的张量详细信息。   未找到有关张量的所需信息的张量不会添加到列表中。这包括没有名称的临时张量。

     

返回：包含张量信息的字典列表。