tf.lite.experimental.Analyzer
Предоставляет набор инструментов для анализа моделей TFLite.
Используется в блокнотах
| Используется в руководстве |
|---|
Пример:
model = tf.keras.applications.MobileNetV3Large() fb_model = tf.lite.TFLiteConverterV2.from_keras_model(model).convert() tf.lite.experimental.Analyzer.analyze(model_content=fb_model) # === TFLite ModelAnalyzer === # # Your TFLite model has ‘1’ subgraph(s). In the subgraph description below, # T# represents the Tensor numbers. For example, in Subgraph#0, the MUL op # takes tensor #0 and tensor #19 as input and produces tensor #136 as output. # # Subgraph#0 main(T#0) -> [T#263] # Op#0 MUL(T#0, T#19) -> [T#136] # Op#1 ADD(T#136, T#18) -> [T#137] # Op#2 CONV_2D(T#137, T#44, T#93) -> [T#138] # Op#3 HARD_SWISH(T#138) -> [T#139] # Op#4 DEPTHWISE_CONV_2D(T#139, T#94, T#24) -> [T#140] # ...
Методы
analyze
@staticmethod
analyze(
model_path=None, model_content=None, gpu_compatibility=False, **kwargs
)
Анализирует указанную модель tflite_model и выводит структуру модели.
Этот инструмент помогает понять flatbuffer-модель TFLite пользователя, выводя внутреннюю структуру графа. Он также предоставляет дополнительные возможности, например проверку совместимости с делегатом GPU.
| Аргументы | |
|---|---|
model_path | Путь к flatbuffer-модели TFLite. |
model_content | Объект flatbuffer-модели TFLite. |
gpu_compatibility | Нужно ли проверять совместимость с делегатом GPU. |
**kwargs | Экспериментальные именованные аргументы API analyze. |
| Возвращаемое значение | |
|---|---|
| Выводит отчет об анализе в консоль. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/lite/experimental/Analyzer