gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-02 06:18:46 26 4
gpt4 key购买 nike

我正在尝试使用 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 文件夹中的模型(初始模型)相同的格式

最佳答案

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

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

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

关于java - 如何在 Tensorflow 移动 Android 应用程序上使用预训练模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55848864/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com