java - 如何在 Tensorflow 移动 Android 应用程序上使用预训练模型？

Question

我正在尝试按照本教程使用 Android 上的张量流移动演示应用程序进行图像检测和分类。我认为该应用程序正在使用初始模型，并且我想使用此链接中的另一个预训练模型“ssd_mobilenetv2_oidv4” 。我在机器学习方面没有太多经验，但我想做的是用 mobilenetv2 模型替换应用程序中使用的模型，因为它接受了更多类的训练。

我试图将预训练模型中的文件复制到应用程序中的资产文件夹并调整应用程序中的路径，但问题是 2 个模型具有不同的格式（我认为），因此应用程序在启动之前就崩溃了.

资产文件夹中的文件（初始模型）： imagenet_comp_graph_label_strings.txt multibox_location_priors.txt tensorflow_inception_graph.pb tiny-yolo.pb

MObilenetV2 我要使用的模型中的文件： checkpoint frozen_inference_graph.pb model.ckpt.data-00000-of-00001 model.ckpt.index model.ckpt.meta pipeline.config and folder "saved_model" containing: a- saved_model.pb b- an empty folder named "variables"

那么如何将我想要使用的模型转换为与资产文件夹中的模型相同的格式（初始模型）

标签： javaandroidtensorflowmachine-learningyolo

Answer 1

这里的问题是您正在尝试使用不同的模型格式。为了能够使用模型动物园检测存储库中的模型 您应该遵循此链接中的 TensorFlow Lite 示例 也不要忘记，如果您不使用 tflite 模型，您应该将模型转换为 tflite。例如，您可以使用本教程