Невозможно вывести результаты с использованием модели обнаружения объектов tflite

Я успешно преобразовал модель ssd_mobilenet_v3 в .tflite, переобучив модель. (ранее обучался на наборе данных coco)

Но при использовании этой модели для обнаружения объектов необходимо выполнить вывод на одном изображении, используя следующий код:

interpreter = tf.lite.Interpreter(model_path)

Я получаю следующую ошибку:

ValueError                                Traceback (most recent call last)
<ipython-input-15-e1c9008b610c> in <module>
----> 1 interpreter = tf.lite.Interpreter("/home/sushanth/Documents      /nuts_poc/tflite_od/nam_model_quantized.lite")
  2 interpreter.allocate_tensors()
  3 input_details = interpreter.get_input_details()
  4 output_details = interpreter.get_output_details()
  5 input_tensor_index = interpreter.get_input_details()[0]["index"]

~/.local/lib/python3.7/site-packages/tensorflow/lite/python/interpreter.py   in __init__(self, model_path, model_content)
 75       self._interpreter = (
 76             _interpreter_wrapper.InterpreterWrapper_CreateWrapperCPPFromFile(
---> 77               model_path))
 78       if not self._interpreter:
 79         raise ValueError('Failed to open {}'.format(model_path))

ValueError: Op builtin_code out of range: 117. Are you using old TFLite binary with newer model?Registration failed.

Пожалуйста, объясните ошибку и возможное решение.

Версия Tensorflow: 1.1.4

ОС: Ubuntu 18.04

Python: 3.7

PS: Я преобразовал модель классификатора (inception_v2) в tflite и использовал приведенный выше код ("интерпретатор = tf.lite.Interpreter (model_path)") без каких-либо ошибок!


person Sushanth    schedule 29.10.2019    source источник
comment
Вы пробовали обновить Tensorflow до 2.0? pip3 install tensorflow==2 Это помогло мне.   -  person Guilherme Vogt    schedule 04.11.2019


Ответы (1)


Обновите версию tensorflow до> = 2.0.0

person Chen Xiaofeng    schedule 02.03.2020