Tensorflow在离线中使用预训练模型

打击

我正在使用Tensorflow的mobilenet和以下代码。当我从本地PC运行此代码时,它将下载重量文件。但是,上传到服务器后,我无法在线下载该文件。

  1. 有什么办法可以将MobileNetV2与tensorflow 2.0一起使用?

  2. 或者,我有自己的配重文件,并且受MobielNetV2培训,那么可以在下面使用没有配重的选项

    mobilenet = tf.keras.applications.mobilenet_v2.MobileNetV2(input_shape =(224,224,3),include_top = False,weights ='imagenet')

  3. 我想从Linux服务器使用supervisor。看来我需要使用绝对路径而不是相对路径。谁能帮助我在脱机环境中与主管一起使用MobileNetV2

巴希尔·卡齐米

您可以先使用:

mobilenet = tf.keras.applications.mobilenet_v2.MobileNetV2(input_shape=(224, 224, 3), include_top=False, weights='imagenet')

然后将权重保存在某处:

mobilenet.save_weights('somefolder/mobilenetweights.h5')

然后,当您离线时,您可以先致电:

mobilenet = tf.keras.applications.mobilenet_v2.MobileNetV2(input_shape=(224, 224, 3), include_top=False, weights=None)

注意Noneinweights参数。之后,您可以从以前保存的文件中加载权重:

mobilenet.load_weights('somefolder/mobilenetweights.h5')

这应该工作。

调用mobilenet模型时,如果要求使用来给它提供imagenet权重weights='imagenet',则需要Internet连接才能下载这些权重。因此它无法离线运行。此处说明的方法应使其起作用。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

在预训练的方式Keras,Tensorflow中使用机器学习模型

如何在 Tensorflow 中使用预训练模型?

如何在 Tensorflow 中使用没有类的预训练模型?

Tensorflow负载预训练模型使用不同的优化器

如何使用预训练的 tensorflow 模型对图像进行预测?

如何在Python中使用预训练的CNN模型

在keras模型中使用预训练的单词嵌入?

Tensorflow:如何在新图中使用预训练权重?

在Tensorflow中使用预训练的inception_resnet_v2

如何在 tensorflow 中使用预训练的对象检测?

使用Tensorflow和Inception V3预训练模型训练高清图像

对 mnist 数据集使用预训练模型

如何使用Tensorflow Keras API从预训练的模型复制特定的图层权重?

如何使用tensorflow.js以角度正确地上传预训练的模型?

TensorFlow对象检测API:使用预训练模型在训练中更改班级数量时的分类权重初始化

使用预训练的(Tensorflow)CNN提取特征

如何在 doc2vec 模型中使用预训练的 word2vec 向量?

使用队列Tensorflow训练模型

建议如何使用Yolov4训练预训练模型?

在TensorFlow中使用预训练的单词嵌入(word2vec或Glove)

Tensorflow:下载并运行预训练的VGG或ResNet模型

来自两个预训练的Tensorflow模型的合并预测

Tensorflow:导入预训练模型(mobilenet,.pb,.ckpt)

将预训练的词嵌入加载到 Tensorflow 模型中

预训练的Inception v3模型的图层名称(tensorflow)

预训练的Tensorflow模型RGB-> RGBY通道扩展

Tensorflow:确定预训练的CNN模型的输出步幅

我如何结合 Dialogflow、Tensorflow 预训练模型和 Flask?

如何使用Node.js将经过转换的预训练keras模型加载到Tensorflow.js中?