属性错误：模块"张量流"没有属性"整个文件读取器"

Question

我一直在尝试实现这段代码，并意识到TensorFlow V2中有很多不推荐使用和更新的代码。代码

filenames=['images/000001.jpg','images/000002.jpg','images/000003.jpg','images/000004.jpg']
labels=[1,0,1,0]
filename_queue=tf.train.string_input_producer(filenames)
reader=tf.WholeFileReader()
filename, content = reader.read(filename_queue)
images=tf.image.decode_jpeg(content, channels=3)
images=tf.cast(images, tf.float32)
resized_images=tf.image.resize_images(images, (224, 224))

我修改了tf.train.string_input_producer(filenames)部分。但我找不到tf.WholeFileReader()的任何替代方案。此外，我正在使用谷歌colab，因此不能依赖V1，因为它说colab不支持TensorFlow V2。这是我的一段代码。

我的代码

import os
import numpy as np
image_path = '/content/drive/MyDrive'
categ = ["IMG"]
for c in categ:
path = os.path.join(image_path, c)
for img in os.listdir(path):
image = [os.path.join(path, img)]


queue=tf.data.Dataset.from_tensor_slices(image)
reader=tf.WholeFileReader() #AttributeError

有人能帮忙吗？

Answer 1

您可能应该能够在这里找到它：

tf.io.WholeFileReader()

而不是

tf.WholeFileReader()