无法使用tflite对象检测模型推断结果

时间:2019-10-29 15:00:56

标签: python tensorflow object-detection tensorflow-lite mobilenet

我通过重新训练将该模型成功地将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,并使用了上面的代码(“ interpreter = tf.lite.Interpreter(model_path)”)!

1 个答案:

答案 0 :(得分:1)

将Tensorflow版本更新为> = 2.0.0