gpt4 book ai didi

python - TensorFlow:将输入发送到导出的模型

转载 作者:行者123 更新时间:2023-12-01 09:32:46 24 4
gpt4 key购买 nike

场景:我导出了 TF 模型,并且有 .pb 文件。该模型是一个经过图像训练的简单分类器。该模型以图像作为输入,并提供图像是否属于某个类别的概率分数。

我正在使用 python 并使用下面的方法加载模型。graph = load_graph('model.pb')

但无法找到一种方法来发送图像作为输入并检索结果。我应该使用哪些属性来发送输入和接收结果?

最佳答案

在 tensorflow 中有几种方法可以做到这一点。最简单的方法是使用 feed_dict。您首先需要从模型中找到输入张量,output_tensor,然后使用 feed dict 提供新输入并获取新输出示例:

    input_place_holder = model.input_place_holder
out_put = model.output_tensor
current_input = your_img_input
result = model.session.run([out_put], feed_dict={input_place_holder: current_input})

将输入输入模型的更高级方法是使用 tensorflow 服务(具有更好的性能等)

关于python - TensorFlow:将输入发送到导出的模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49816542/

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