调整输入尺寸后，tensorflow lite（tflite）调用错误

Question

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

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

调用时会报错：

interpreter->AllocateTensors()

如果我注释掉解释器->ResizeInputTensor(input, sizes);那么一切都很好。

有什么建议吗？

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

Answer 1

ResizeInputTensor 受神经网络架构的限制。它失败了，因为 MobileNet 和 MobileNet SSD 只能处理固定大小的输入。

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