tensorflow lite（tflite）在调整输入维度大小后调用错误

Question

我使用的是mobilenet_ssd.tflite作为官方tensorflow github的模式。代码如下：

int input = interpreter->inputs()[0];
interpreter->ResizeInputTensor(input, sizes);

这会在调用时导致错误：

interpreter->AllocateTensors()

如果我注释掉解释器 - ＆gt; ResizeInputTensor（输入，大小）;然后一切都很好。

有什么建议吗？

我问的另一个问题是： change the input image size for mobilenet_ssd using tensorflow

Answer 1

ResizeInputTensor受神经网络架构的限制。它自从MobileNet＆amp; MobileNet SSD只能处理固定大小的输入。

可能有用的是改变批量大小。例如，您可以尝试将大小从（1,244,244,3）更改为（4,244,244,3），并在一次Invoke调用中对4个图像进行推断。