Extend 中不兼容的 GraphDef 版本
Incompatible GraphDef versions in Extend
我有一些代码可以创建一个图形来处理一些图像,然后在循环中迭代 sess.run()
以获取形状为 [*, 299, 299, 3] 的图像张量批次。然后我想将这些图像输入初始模型。
所以,我添加了一些代码来加载初始模型:
def create_graph():
""""Creates a graph from saved GraphDef file and returns a saver."""
# Creates graph from saved graph_def.pb.
print 'Loading graph...'
with tf.Session() as sess:
with gfile.FastGFile('/web/tensorflow_transfer/resources/classify_image_graph_def.pb', 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
_ = tf.import_graph_def(graph_def, name='')
return sess.graph
g = create_graph()
for i in range(training_steps):
sess.run(...)
现在我在 运行 run()
:
时收到此错误
File "/usr/local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 368, in run
results = self._do_run(target_list, unique_fetch_targets, feed_dict_string)
File "/usr/local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 420, in _do_run
raise RuntimeError(compat.as_text(tf_session.TF_Message(status)))
RuntimeError: Incompatible GraphDef versions in Extend: 1 != 0
这很可能是因为使用的 TensorFlow 版本太旧而无法读入,并且 运行 图表 -- 该图表是使用较新版本的 GraphDef 创建的。尝试升级到 0.7 或 HEAD,然后再次 运行 您的代码。
我有一些代码可以创建一个图形来处理一些图像,然后在循环中迭代 sess.run()
以获取形状为 [*, 299, 299, 3] 的图像张量批次。然后我想将这些图像输入初始模型。
所以,我添加了一些代码来加载初始模型:
def create_graph():
""""Creates a graph from saved GraphDef file and returns a saver."""
# Creates graph from saved graph_def.pb.
print 'Loading graph...'
with tf.Session() as sess:
with gfile.FastGFile('/web/tensorflow_transfer/resources/classify_image_graph_def.pb', 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
_ = tf.import_graph_def(graph_def, name='')
return sess.graph
g = create_graph()
for i in range(training_steps):
sess.run(...)
现在我在 运行 run()
:
File "/usr/local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 368, in run
results = self._do_run(target_list, unique_fetch_targets, feed_dict_string)
File "/usr/local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 420, in _do_run
raise RuntimeError(compat.as_text(tf_session.TF_Message(status)))
RuntimeError: Incompatible GraphDef versions in Extend: 1 != 0
这很可能是因为使用的 TensorFlow 版本太旧而无法读入,并且 运行 图表 -- 该图表是使用较新版本的 GraphDef 创建的。尝试升级到 0.7 或 HEAD,然后再次 运行 您的代码。