我使用的是一个图形文件(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