Квантование тензорного потока в Windows

1

Я заморозил свою модель и получил файл.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?

  • 0
    TensorFlow Lite все еще является экспериментальным API, и поддержка Windows, кажется, не стабильна (если вообще), см. # 16374 или # 14607 . Вы используете последнюю версию TensorFlow? Если нет, обновление может помочь.
Теги:
tensorflow
quantization

1 ответ

0

toco в настоящее время не поддерживается в Windows build для cmake. Это то, что я когда-то читал.

Ещё вопросы

Сообщество Overcoder
Наверх
Меню