如何在Tensorflow移动android应用程序上使用预训练模型?

问题描述 投票:-1回答:1

我试图在Android上使用张量流移动演示应用程序进行图像检测和分类,这是在tutorial之后。我认为该应用程序正在使用初始模型,我想使用另一个预先训练的模型“ssd_mobilenetv2_oidv4”来自这个link。我没有很多机器学习的经验,但我想做的是用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"

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

java android tensorflow machine-learning yolo
1个回答
0
投票

请注意,在还原检查点文件时,您不必事先初始化它们。

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

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

© www.soinside.com 2019 - 2024. All rights reserved.