在带有图像的android中使用tensorflow lite解释器

Question

所以，我在android平台上部署模型时遇到了一个痛点。 我已经在 python 中训练了一个模型并将其转换为适用于 android 平台的 tensorflow lite 格式。 问题是如何做我在 Java 中的 python 中所做的事情/处理。 在网上进行了繁琐的搜索后，没有。的例子，没有什么能清楚地解释这个过程。

例如：如何在 Java 中模拟预处理/规范化算法，即对图像进行预处理和后处理。

如果使用量化模型进行推理以及如何实现将多个输入传递给模型，代码中会发生哪些变化。

下面是python中对应的代码：

def preprocess(img):
  return (img / 255. - 0.5) * 2

def deprocess(img):
  return (img + 1) / 2

谁能指导我如何在 Java 中和多个输入中做同样的事情。 对于在 Android 中寻找模型集成的人来说，这可能是一个圣杯。

提前致谢！！！

Answer 1

请查看TensorFlow Lite Android Support Library。

特别是，深入了解TensorProcessor 及其supported ops，其中包括量化/去量化操作（您在示例代码中称为预处理/去处理）。

另外，这是一个full example Android project using TFLite and the Support Library。