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

Question

我正在尝试使用 Android 上的 tensorflow 移动演示应用程序进行图像检测和分类，如下 tutorial 。我认为该应用程序正在使用初始模型，我想使用另一个预训练模型“ssd_mobilenetv2_oidv4 ” link 。我在机器学习方面没有太多经验，但我想做的是将应用程序中使用的模型替换为 mobilenetv2 模型，因为它接受了更多类别的训练。

我尝试将预训练模型中的文件复制到应用程序中的 Assets 文件夹中，并调整应用程序中的路径，但问题是这两个模型具有不同的格式(我认为)，因此应用程序崩溃之前它甚至开始了。

assets文件夹中的文件(inception模型): imagenet_comp_graph_label_strings.txtmultibox_location_priors.txttensorflow_inception_graph.pbtiny-yolo.pb

MObilenetV2 我想使用的模型中的文件:检查点frozen_inference_graph.pbmodel.ckpt.data-00000-of-00001model.ckpt.indexmodel.ckpt.meta管道.config和文件夹“saved_model”包含:a-saved_model.pbb- 一个名为“variables”的空文件夹

那么如何将我想要使用的模型转换为与 Assets 文件夹中的模型(初始模型)相同的格式

Answer 1

请注意，当您恢复检查点文件时，您不必事先初始化它们。

tf.train.Saver 还可以恢复(tf.train.Saver.restore):

https://www.tensorflow.org/api_docs/python/tf/train/Saver#restore