web-dev-qa-db-fra.com

Comment charger un modèle tflite en script?

J'ai converti le .pb fichier vers tflite fichier en utilisant le bazel. Maintenant, je veux charger ce modèle tflite dans mon script python juste pour tester que la météo me donne une sortie correcte ou non?

7
Harshit Mishra

Vous pouvez utiliser TensorFlow Lite Python interpreter pour charger le modèle tflite dans un python = Shell, et testez-le avec vos données d'entrée.

Le code sera comme ceci:

import numpy as np
import tensorflow as tf

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

# Get input and output tensors.
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()

# Test model on random input data.
input_shape = input_details[0]['shape']
input_data = np.array(np.random.random_sample(input_shape), dtype=np.float32)
interpreter.set_tensor(input_details[0]['index'], input_data)

interpreter.invoke()

# The function `get_tensor()` returns a copy of the tensor data.
# Use `tensor()` in order to get a pointer to the tensor.
output_data = interpreter.get_tensor(output_details[0]['index'])
print(output_data)

Le code ci-dessus provient du guide officiel de TensorFlow Lite, pour plus d'informations, lisez ceci .

23
Jing Zhao