仅在Android中使用Tensorflow .pb文件就足够了

Question

我正在研究图像分类，我需要一些建议。我正在使用带有后端tensorflow的keras构建大型图像分类器，因为我想将该模型用于Android应用程序，因此我对其进行了训练，并将这些检查点转换为protobuf（.pb）文件。在许多示例中，我看到2个文件，当它们导出到android资产文件夹时，分别是protobuf文件和label.txt文件，因此可以只在android上使用protobuf文件吗？   谢谢！

Answer 1

如果要在Android上使用该模型，则需要先将其转换为TensorFlow Lite。这个codelab可能会有所帮助。该模型将返回索引和概率。如果要将数字索引转换为标签，则需要该映射的labels.txt文件。

或者，您可以使用AutoML in ML Kit建立图像分类模型。客户端库负责自动完成该映射。