python 解析pb文件_Tensorflow:从图形文件(.pb文件)中获取预测

我使用的是一个图形文件(pb文件),这个张量流模型的目的是提供对特定图像的预测

我开发了一个代码,加载了图形文件,但我不能统计会话。

可用文件包括:培训模式已保存_型号.pb在

变量

训练模型变量_变量.数据-00000-of-00001

训练模型变量_变量.索引在

输出错误包含一个大的模型列表层。什么我可以在这种情况下,任何帮助都是感激的

这是我用来加载/运行模型的代码import tensorflow as tf

import sys

import os

import matplotlib.image as mpimg

import matplotlib.pyplot as plt

from tensorflow.core.protobuf import saved_model_pb2

from tensorflow.python.util import compat

from tensorflow.python.platform import gfile

export_dir = os.path.join("./", "variables/")

filename = "imgpsh_fullsize.jpeg"

raw_image_data = mpimg.imread(filename)

g = tf.Graph()

with tf.Session(graph=g) as sess:

model_filename ='training_model_saved_model.pb'

with gfile.FastGFile(model_file

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值