Tensorflow Lite tflite流程以退出代码-1073741819完成

时间:2019-06-29 16:33:31

标签: python tensorflow-lite

发布一个让我花了一段时间才能发现并且在搜索时找不到的问题,希望这可以帮助其他人节省一些时间。

调用set.tensor程序时崩溃,并出现响应,进程退出代码为-1073741819

interpreter = tf.lite.Interpreter(model_path="....")
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()
interpreter.set_tensor(input_details[0]['index'],image_for_input )
interpreter.invoke()

1 个答案:

答案 0 :(得分:0)

在调用set_tensor之前必须先调用allocate_tensors

interpreter = tf.lite.Interpreter(model_path="....")
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()

interpreter.allocate_tensors()

interpreter.set_tensor(input_details[0]['index'],image_for_input )
interpreter.invoke()