tensorflow 只支持 300x300 的图片尺寸？

Question

请问tensorflowlite输入图片只支持300 x 300吗？ 那么，除了 300x300 输入图像之外，任何其他输入都不会被 tensorflowlite 识别>？

Answer 1

TL;DR

不，Tensorflow-Lite 没有 300x300 的限制 :)

Tensorflow-Lite 支持您想要的任何图像尺寸 - 您可能会受到平台硬件的限制，然后是 TFL 库。

这是example from TFL：

inputImageBuffer = loadImage(bitmap, sensorOrientation);
tflite.run(inputImageBuffer.getBuffer(), outputProbabilityBuffer.getBuffer().rewind());

这里，“输入图像”是您在loadImage 中看到的Android 的bitmap。 该位图是可以在预处理中配置的图像，但绝对不限于300x300。

更多信息

在 Tensorflow-Lite (TFL) 中，推理（您向其提供输入并从中接收输出的实例）使用字节缓冲区来执行。

我认为当您询问图像尺寸时，您指的是称为 TensorImage 的字节缓冲区的概念。

在创建模型并将输入参数设置为“图像”时，您也在创建包含图像高度和宽度的元数据。

您可以使用 TFL 库从模型中获取图像尺寸，因此在这种情况下您会受到模型的限制。

有关更多信息和示例代码，我建议您查看以下内容：

• 对于使用 TFL 运行的实例：https://www.tensorflow.org/lite/guide/inference
• 代码示例：https://www.tensorflow.org/lite/examples
• 对于支持库（您也可以使用它与图像尺寸进行交互）：https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/experimental/support/java/README.md