列出 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方法获取Tensors列表

interpreter.get_tensor_details()

根据文档，

Gets tensor details for every tensor with valid tensor details. Tensors where required information about the tensor is not found are not added to the list. This includes temporary tensors without a name.

Returns: A list of dictionaries containing tensor information.

Answer 2

您可以使用可视化脚本获取所有使用的 Tensorflow Lite 操作的列表。

wget -O tflite_visualize.py https://raw.githubusercontent.com/tensorflow/tensorflow/master/tensorflow/lite/tools/visualize.py

假设您的模型保存在 model.tflite 使用下载的脚本创建 html 文件。

python tflite_visualize.py model.tflite model_visualization.html

就在标记为 Ops 的部分中。

列出 TFLite 模型的操作

List operations of a TFLite model

tensorflow

tensorflow-lite