在android中使用tensorflow lite解释器显示图片

Question

因此，我在android平台上遇到了模型部署的一个痛点。我在python中训练了一个模型，并将其转换为适用于android平台的tensorflow lite格式。问题是如何做我用python在Java中做的事情/处理。在网上繁琐地搜索了一遍之后，没有。例如，没有任何东西真正清楚地解释了这个过程。

例如:如何在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。