TensorFlow2.0.0 Alpha- 模块 'tensorflow' 没有属性 'gfile'

Question

在Windows10上运行以下Python代码时，我遇到了上述错误，Python 3.7.0和tensorflow==2.0.0-alpha0：

import numpy as np
import os
import six.moves.urllib as urllib
import tarfile
import tensorflow as tf
import zipfile
from collections import defaultdict
from io import StringIO
from matplotlib import pyplot as plt
from PIL import Image
from object_detection.utils import ops as utils_ops
%matplotlib inline
from object_detection.utils import label_map_util
from object_detection.utils import visualization_utils as vis_util
PATH_TO_LABELS = os.path.join('C:\Users\User\models\research\object_detection\data', 'mscoco_label_map.pbtxt')
PATH_TO_FROZEN_GRAPH = os.path.join('C:\Users\User\models\research\object_detection\ssd_mobilenet_v1_coco_2017_11_17', 'frozen_inference_graph.pb')
PATH_TO_CKPT = PATH_TO_FROZEN_GRAPH
with detection_graph.as_default():
od_graph_def = tf.compat.v1.GraphDef()
with tf.compat.v2.io.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='')
NUM_CLASSES=90
label_map = label_map_util.load_labelmap(PATH_TO_LABELS)

有什么建议吗？ 请 吉拉德

Answer 1

gfile在tensorflow-2中移至tf.io。 所以你需要导入tf.io而不仅仅是导入tensorflow as tf

Answer 2

在 tf2.0 中，gfile 包已移至 tf.io 中。 所以你可以使用 tf.io.gfile