Tensorflow Lite值错误：无法设置张量：维度不匹配

Question

这可能是个愚蠢的问题，但我对机器学习和Tensorflow还是新手。 我正在尝试使用Tensorflow Lite在Raspberry Pi上运行对象检测API。我试图用这个例子来修改我的代码

https://github.com/freedomtan/tensorflow/blob/deeplab_tflite_python/tensorflow/contrib/lite/examples/python/object_detection.py

这段代码将从图像中检测对象。但我想通过Pi摄像机实时检测物体，而不是图像。 我试着修改这个代码来读取摄像机的输入而不是图像。这是我的密码-

^{pr2}$

但是我得到了一个错误-

Traceback (most recent call last):
  File "New_object_detection.py", line 257, in <module>
    interpreter.set_tensor(input_details[0]['index'], image_np_expanded)
  File "/home/saurabh/.local/lib/python3.6/site-packages/tensorflow/contrib/lite/python/interpreter.py", line 151, in set_tensor
    self._interpreter.SetTensor(tensor_index, value)
  File "/home/saurabh/.local/lib/python3.6/site-packages/tensorflow/contrib/lite/python/interpreter_wrapper/tensorflow_wrap_interpreter_wrapper.py", line 133, in SetTensor
    return _tensorflow_wrap_interpreter_wrapper.InterpreterWrapper_SetTensor(self, i, value)
ValueError: Cannot set tensor: Dimension mismatch

有谁能告诉我如何修复这个错误或者建议一个教程吗？在

Answer 1

许多基于图像的机器学习模型都是用固定大小的输入进行训练的。原始图像可能具有不同的尺寸，但会调整为固定大小（例如224x224x3）。在

因此，在将输入输入输入到模型之前，需要调整图像的大小。它可能工作得很好，因为训练数据也可以从不同的大小调整大小。在

正如上面的评论已经指出的，cv.resize可以做到这一点。在