runtime error python path_python-TensorFlow RuntimeError:在SavedModel中找不...

当我使用simple_save保存模型时,尝试加载模型时出现运行时错误.

要保存的代码是:

session = Session()

inputs = tf.placeholder(dtype=tf.float32, shape=(None, height, width, in_channel_size), name='input_img')

model = Some_Model(inputs, num_classes=no_of_defects, is_training=False)

logits, _ = model.build_model()

predictor = tf.nn.softmax(self.logits, name='logits_to_softmax')

feed_dict = {inputs: inputs}

prediction_probabilities = session.run(self.predictor, feed_dict=feed_dict)

tf.saved_model.simple_save(self.session, path,

inputs={"inputs" : self.inputs},

outputs={"predictor": self.predictor})

要加载的代码是:

tf.saved_model.loader.load(session, tag_constants.SERVING, path)

这给出了错误:

RuntimeError: MetaGraphDef associated with tags serve could not be found in SavedModel. To inspect available tag-sets in the SavedModel, please use the SavedModel CLI: `saved_model_cli`

当我跑步

saved_model_cli show --dir path --tag_set serve --signature_def serving_default

我懂了

The given SavedModel SignatureDef contains the following input(s):

inputs['inputs'] tensor_info:

dtype: DT_FLOAT

shape: (-1, 512, 1024, 8)

name: input_img:0

The given SavedModel SignatureDef contains the following output(s):

outputs['predictor'] tensor_info:

dtype: DT_FLOAT

shape: (-1, 512, 1024, 25)

name: logits_to_softmax:0

Method name is: tensorflow/serving/predict

我究竟做错了什么?

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值