import tensorflow as tf
import numpy as np
#Load your TFLite model.
TF_LITE_MODEL_FILE_NAME = "Your_TFLite_file.tflite"
interpreter = tf.lite.Interpreter(model_path=TF_LITE_MODEL_FILE_NAME)
#Gives you input and output details
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()
#Gives you all tensor details index-wise. you will find all the quantization_parameters here
interpreter.get_tensor_details()
#get individual tensor value. interpreter.get_tensor(give_index_number). You will find the index for individual tensor index from get_tensor_details
interpreter.allocate_tensors()
r= interpreter.get_tensor(12).astype(np.float32)
print('Tensors', r)
1条答案
按热度按时间swvgeqrz1#
有几种方法可以提取有关权重、比例和零点值的信息。
第一种方式
您还可以从TensorFlow网站找到有关以下代码的其他信息。
方法二(简单方法):
将您的TFLite文件上传到Netron Website上。在那里您可以得到关于您的TFLite文件的很多信息。您也可以在您的PC上安装Netron。这里是在您的PC上安装Netron的Netron git link。