Я заморозил свою модель и получил файл.pb. Затем я квантую свою модель, используя tocoConverter в Linux, поскольку она не поддерживается в Windows. У меня есть quantized_model.tflite. Я могу загрузить его и получить прогнозы в Linux, но у меня есть проблемы, связанные с Windows, как того требует мой проект. Я попытался загрузить его, используя tf.contrib.lite.Interpreter, используя этот код:
import numpy as np
import tensorflow as tf
# Load TFLite model and allocate tensors.
interpreter=tf.contrib.lite.Interpreter(model_path="quantized_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']
# change the following line to feed into your own data.
input_data = np.array(np.random.random_sample(input_shape), dtype=np.float32)
interpreter.set_tensor(input_details[0]['index'],input_data)
interpreter.invoke()
output_data = interpreter.get_tensor(output_details[0]['index'])
print(output_data)
*ImportError: No module named 'tensorflow.contrib.lite.python.Interpreter*
Но это не удалось с "Нет модуля с именем" tensorflow.contrib.lite.python.interpreter ". Я всегда получаю эти ошибки в Windows, пытаясь использовать что-то из tf.contrib.lite. Может быть, есть способ загрузить это на Windows? Или вы можете предложить альтернативные варианты квантования модели в Windows?
toco в настоящее время не поддерживается в Windows build для cmake. Это то, что я когда-то читал.