Я заморозил свою модель и получил файл .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*
Но не удалось с ошибкой «Нет модуля с именем tennsflow.contrib.lite.python.interpreter».Я всегда получаю эти ошибки в Windows, когда пытаюсь использовать что-то из tf.contrib.lite.Может быть, есть способ загрузить это на Windows?Или вы можете посоветовать альтернативные варианты для квантования модели в Windows?