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

Question

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

Answer 1

TL;博士

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

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

这是TFL 的一个例子：

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

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

更多信息

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

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

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

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

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

• 对于使用 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