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

Question

使用 tensorflow==2.0.0-alpha0 在 Windows10 上使用 Python 3.7.0 运行以下 Python 代码时出现上述错误：

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移至tf.ioin tensorflow-2。所以你需要导入tf.io而不是仅仅导入tensorflow as tf

Answer 2

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