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

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 一起使用