我试图在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文件夹(初始模型)中的模型相同的格式
请注意,在还原检查点文件时,您不必事先初始化它们。
tf.train.Saver
也可以恢复(tf.train.Saver.restore
):
https://www.tensorflow.org/api_docs/python/tf/train/Saver#restore