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

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

我在 Google Colab 上使用 tensorflow 2.0 训练了一个简单的 mnist 模型并将其保存在 .json格式。 Click here 查看我编写代码的 Colab Notebook。然后在 运行 命令

!simple_tensorflow_serving --model_base_path="/" --model_platform="tensorflow"

显示错误 AttributeError: module 'tensorflow' has no attribute 'gfile'

simple_tensorflow_serving 有助于轻松地将经过训练的 tensorflow 模型部署到生产中。

版本 我正在使用:

(1) TensorFlow - 2.0

(2) simple_tensorflow_serving - 0.6.4

提前谢谢你:)

Simple Tensorflow Serving 还没有为 Tensorflow 2.0 做好准备,因为它使用的是旧的 API。 在 Tensorflow 2.0 中,gfile 包已移至 tf.io.

然后,您必须使用 Simple Tensorflow Serving 将您的 Tensorflow 实例降级到 TF 1.13

In 2.0, tf.gfile.* is replaced by tf.io.gfile.*.

当我遇到错误时:

  File "/Users/MRJ/anaconda3/envs/python37-tf2.1/lib/python3.7/
site-packages/object_detection/utils/label_map_util.py",
line 137, in load_labelmap

with tf.gfile.GFile(path, 'r') as fid:
  AttributeError: module 'tensorflow' has no attribute 'gfile'

1.Find label_map_util.py 第 137 行。

2.Replace tf.gfile.GFiletf.io.gfile.GFile

对我有用。

tensorflow issue #31315

tf.gfile.GFile更改为tf.io.gfile.GFile 它适用于此实例但不适用于所有其他文件 示例 tf.io.gfile.FastGFile 抛出错误提示“AttributeError:模块 'tensorflow._api.v2.io.gfile' 没有属性 'FastGFile'”。我希望它能节省你的时间。

相反,尝试

tf.io.gfile.GFile(
    name, mode='r'
)

即“.io”将解决您的所有问题,而不是降级您的 tf