如何在树莓派上运行使用 GCP 训练的 TFLITE 模型

Question

我使用 Google Cloud 平台 AutoML 训练了一个模型，该模型将垃圾图像分类为“湿”或“干”。使用 GCP，我将经过训练的模型下载为 TFLite 模型并得到：

1. .txt 文件，带有 2 个可能的标签
2. .json 文件
3. .tflite 文件用于实际模型。

鉴于这些文件，我如何在树莓派上运行推理。我进行了相当多的研究，但我无法找到如何做同样的事情。有人可以分享有用的资源或任何类似的相关经验吗？

Answer 1

您是否尝试过 TFLite 中的 Raspberry PI 示例：https://github.com/tensorflow/examples/tree/master/lite/examples/image_classification/raspberry_pi

看来你想做的和那个很相似。