TensorFlow:将输入发送到导出的mod

2024-04-20 03:21:00 发布

您现在位置:Python中文网/ 问答频道 /正文

场景:我已经导出了一个TF模型,还有.pb文件。该模型是一个简单的分类器,经过图像训练。该模型以图像为输入,给出图像是否属于某一类别的概率得分。你知道吗

我正在使用python并使用下面的代码加载模型。 图=加载图('型号.pb')

但无法找到将图像作为输入发送和检索结果的方法。我应该使用哪些属性来发送输入和接收结果?你知道吗


Tags: 文件方法代码模型图像属性分类器tf
1条回答
网友
1楼 · 发布于 2024-04-20 03:21:00

在tensorflow中有几种方法。最简单的方法是使用feed dict,首先需要定位输入张量,从模型中输出张量,然后使用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服务(它具有更好的性能等)

相关问题 更多 >