\'BatchDataset\' 对象没有属性 \'shape\'

Question

这是我的代码：

train_images = tf.keras.utils.image_dataset_from_directory(
  '/content/drive/MyDrive/ArabicHandwritten2/train')

train_labels = tf.keras.utils.image_dataset_from_directory(
  '/content/drive/MyDrive/ArabicHandwritten2/test')

train_images = tf.reshape(train_images.shape[0], 256, 256, 3).astype('float32')
train_images = (train_images - 127.5) / 127.5 # Normalize the images to [-1, 1]

BUFFER_SIZE = 16765
BATCH_SIZE = 32

train_dataset = tf.data.Dataset.from_tensor_slices(train_images).shuffle(BUFFER_SIZE).batch(BATCH_SIZE)

错误是：

AttributeError: 'BatchDataset' object has no attribute 'shape'

我该如何解决错误？
谢谢。

Answer 1

因为image_dataset_from_directory 返回一个没有shape 属性的tf.data.Dataset。你可以看到如何使用tf.reshape函数here：