python-3.x - ValueError:使用 tf.saved_model.load 导入 SavedModel 需要“tags=”参数
问题描述
我正在关注带有 tensorflow 1.14 的 Keras教程集线器。我没有 pip install tf-nightly-gpu 因为它似乎不适用于 MacOS。一切顺利,直到 keras.Sequential() :
import tensorflow_hub as hub
from tensorflow.keras import layers
classifier_url ="https://tfhub.dev/google/imagenet/resnet_v2_50/feature_vector/3"
IMAGE_SHAPE = (224, 224)
classifier = tf.keras.Sequential([
hub.KerasLayer(classifier_url, input_shape=IMAGE_SHAPE+(3,))
])
最后一行给出以下错误:
ValueError: Importing a SavedModel with tf.saved_model.load requires a 'tags=' argument if there is more than one MetaGraph. Got 'tags=None', but there are 2 MetaGraphs in the SavedModel with tag sets [[], ['train']]. Pass a 'tags=' argument to load this SavedModel.
它与 tensorflow 版本有关还是与其他东西有关?
MacOS High Sierra 10.13.1 python 3.6.8 tensorflow 1.14.0
解决方案
这在 TF 1.14.0 中对我有用:
# Image information
HEIGHT = 224
WIDTH = 224
CHANNELS = 3
IMAGE_SHAPE = (HEIGHT, WIDTH)
feature_extractor_url = "https://tfhub.dev/google/imagenet/resnet_v2_50/feature_vector/3" #@param {type:"string"}
module = hub.Module(feature_extractor_url, tags=['train'])
feature_extractor_layer = hub.KerasLayer(module,
input_shape=(HEIGHT, WIDTH, CHANNELS))
推荐阅读
- java - http 状态 404 - 在 jsp 中找不到,Tomcat 服务器上的 servlet
- javascript - 编写天蓝色函数以在javascript中将pdf转换为图像?
- r - 问题按降序排序R数据框
- c# - 有没有办法转换为泛型函数中传递的类型?
- r - R:作为数据框的一部分列出
- android - 资源的重复值
- flutter - 在 Flutter 中使用具有自定义类值的 FontAwesomeIcons
- django - ModuleNotFoundError:升级到 python 3 时没有名为“textencoder”的模块错误
- c - 在控制台中移动光标和打印字符不适用于睡眠
- python - 处理字典中的列表