通过 tflite 模型传递单个图像进行图像分类

Posted 2023-02-16

【中文标题】通过 tflite 模型传递单个图像进行图像分类

【英文标题】：Image classification by passing single images through tflite model

【发布时间】：2022-01-12 07:54:33

【问题描述】：

我一直在按照本教程创建一个 tflite 模型。https://www.tensorflow.org/lite/tutorials/model_maker_image_classification 我改变的一件事是数据集，我只使用了两个目录来满足我的需要，现在我已经完成了模型的训练和下载，我无法找到传递单个图像的方法（如果相关的话，png 格式）通过新训练的模型。我是新手，我唯一能找到的是 android 平台上的东西，我在 python 中需要它，所以我可以将它部署在树莓派上。 到目前为止，我需要创建一个包含两个类名称的文本文件吗？我真的迷路了，因为到目前为止唯一可行的线路是interpreter = tf.lite.Interpreter(model_path="model.tflite")

【问题讨论】：

如果您传递的图像与用于训练/测试模型的图像格式相同，您可以使用model.predict()

【参考方案1】：

没有任何代码描述很难提供帮助。

只要模型的接口相同，就没有关系。这意味着您的训练模型和您想要预测的图像应该具有相同的张量形状。

您无需为预测创建任何文本文件。您只需要设置输入张量并执行模型。