问调整输入维度大小后，tensorflow lite(tflite)调用错误
EN

Stack Overflow用户

提问于 2018-06-07 15:50:27

回答 1查看 2.3K关注 0票数 0

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

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

这会在调用时导致错误：

interpreter->AllocateTensors()

如果我注释掉interpreter->ResizeInputTensor(input，size)，那么一切都很好。

有什么建议吗？

发布于 2018-06-09 06:20:35

ResizeInputTensor受到神经网络体系结构的限制。它失败了，因为MobileNet和MobileNet固态硬盘只能处理固定大小的输入。

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

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/50735705

复制

相似问题

问调整输入维度大小后，tensorflow lite(tflite)调用错误EN