解释ssd_resnet50对象检测模型的.tflite文件时出错

Question

错误是Kernal Restarting: The kernel appears to have died. It will restart automatically.正在按照here所述的步骤进行。内核在执行interpreter.invoke()时崩溃-在下面的代码中。

我正在使用Tensorflow 1.14.0通过Jupyter Notebook运行代码

编辑1：当我为二进制分类模型的.tflite文件运行相同的代码时，效果很好。

interpreter = tf.lite.Interpreter(model_path=PATH_TO_MODEL)
interpreter.allocate_tensors()

# Get input and output tensors.
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()

# Load image for testing
image = cv2.imread(os.path.join(os.getcwd(),IMAGE_NAME))

input_data_f32 = np.array(image, dtype=np.float32)
input_data = np.expand_dims(input_data_f32, axis=0)
interpreter.set_tensor(input_details[0]['index'], input_data)

interpreter.invoke()

output_data = interpreter.get_tensor(output_details[0]['index'])