Question

我正在使用的对象检测应用程序的一部分中使用以下功能

detection_graph = tf.Graph()
with detection_graph.as_default():
  od_graph_def = tf.GraphDef()
  with tf.gfile.GFile(PATH_TO_FROZEN_GRAPH, 'rb') as fid:
    serialized_graph = fid.read()
    od_graph_def.ParseFromString(serialized_graph)
    tf.import_graph_def(od_graph_def, name='')

运行此代码时，出现以下错误：

AttributeError                            Traceback (most recent call last)
<ipython-input-6-d55b98fd5a78> in <module>
      1 detection_graph = tf.Graph()
      2 with detection_graph.as_default():
----> 3   od_graph_def = tf.GraphDef()
      4   with tf.gfile.GFile(PATH_TO_FROZEN_GRAPH, 'rb') as fid:
      5     serialized_graph = fid.read()

AttributeError: module 'tensorflow' has no attribute 'GraphDef'

我正在使用TensorFlow 2.0.0。这与版本不匹配有关吗？

此代码来自以下link。

Answer 1

是的，这是一个版本不匹配的问题，您不应盲目地假设每个库都支持TensorFlow 2.0（仅是最近发布的版本）。

如果库未公布对TensorFlow 2.0的显式支持，则应将其与TensorFlow 1.x一起使用

Answer 2

确保已正确安装链接中提供的库，

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md

什么是Tensorflow版本2.0.0的'tensorflow'没有属性'GraphDef'？

2 个答案: